Debug

Author: phumthep

src/pownet/core/data_processor.py

@@ -71,7 +71,7 @@ def load_transmission_data(self) -> None:
         # Print files in get_database_dir()
         print("\n\nDEBUGGING")
         print(get_database_dir())
-        print("\nFile list:\n", os.listdir(get_database_dir()))
+        print("File list:\n", os.listdir(get_database_dir()))
 
         self.transmission_params: dict = (
             pd.read_csv(

src/pownet/folder_utils.py

@@ -35,9 +35,7 @@ def get_home_dir() -> str:
 
 
 def get_database_dir() -> str:
-    return os.path.join(
-        os.path.dirname(os.path.dirname(os.path.abspath(__file__))), "database"
-    )
+    return os.path.join(get_pownet_dir(), "src", "database")
 
 
 def get_test_dir() -> str: