fix: data folder unlocated

philsv · Oct 24, 2023 · 85b85d4 · 85b85d4
1 parent a6c3e7e
commit 85b85d4
Show file tree

Hide file tree

Showing 5 changed files with 10 additions and 2 deletions.
diff --git a/pycot/data/contract_names.json → data/contract_names.json b/pycot/data/contract_names.json → data/contract_names.json
diff --git a/pycot/data/cot_reports_data.json → data/cot_reports_data.json b/pycot/data/cot_reports_data.json → data/cot_reports_data.json
diff --git a/pycot/data/format_columns.json → data/format_columns.json b/pycot/data/format_columns.json → data/format_columns.json
diff --git a/pycot/extract_report_data.py b/pycot/extract_report_data.py
@@ -19,15 +19,23 @@ def get_cot_data() -> dict:
     """
     Extracts the Commitment of Traders Reports metadata.
     """
-    with open(BASE_PATH / "pycot/data/cot_reports_data.json", "r") as f:
+    with open(BASE_PATH / "data" / "cot_reports_data.json", "r") as f:
         return json.load(f)
 
 
 def get_formating_data() -> dict:
     """
     Extracts the column formating data.
     """
-    with open(BASE_PATH / "pycot/data/format_columns.json", "r") as f:
+    with open(BASE_PATH / "data" / "format_columns.json", "r") as f:
+        return json.load(f)
+
+
+def get_contract_data() -> dict:
+    """
+    Extracts the contract data.
+    """
+    with open(BASE_PATH / "data" / "contract_names.json", "r") as f:
         return json.load(f)
 
 

diff --git a/pycot/data/__init__.py → tests/__init__.py b/pycot/data/__init__.py → tests/__init__.py