add data to the package

Author: baruch11

MANIFEST.in

@@ -12,4 +12,3 @@ dependencies:
     - pip
     - pip:
       - sphinx-markdown-tables==0.0.17
-      - -e .

environment.doc.yml

@@ -11,7 +11,8 @@
 
 from qolmat.benchmark import missing_patterns
 
-SCRIPT_DIR = os.path.abspath(os.path.dirname(__file__))
+CURRENT_DIR = os.path.abspath(os.path.dirname(__file__))
+ROOT_DIR = os.path.join(CURRENT_DIR, "..")
 
 
 def read_csv_local(data_file_name: str) -> pd.DataFrame:
@@ -27,7 +28,7 @@ def read_csv_local(data_file_name: str) -> pd.DataFrame:
     df : pd.DataFrame
         dataframe
     """
-    df = pd.read_csv(f"{SCRIPT_DIR}/../../data/{data_file_name}.csv")
+    df = pd.read_csv(os.path.join(ROOT_DIR, "data", f"{data_file_name}.csv"))
     return df
 
 

data/beijing.csv renamed to qolmat/data/beijing.csv

@@ -78,6 +78,8 @@
     "Programming Language :: Python :: 3.10",
 ]
 
+PACKAGE_DATA = {"qolmat": ["data/beijing.csv", "data/conductors.csv"]}
+
 setup(
     name=DISTNAME,
     version=VERSION,
@@ -96,4 +98,5 @@
     extras_require=EXTRAS_REQUIRE,
     classifiers=CLASSIFIERS,
     zip_safe=False,
+    package_data=PACKAGE_DATA,
 )