hot fix: need to put entry point in separate dir

so it does not see the datasets so it does not collide with HF datasets
ufal · Nov 5, 2024 · 80a2156 · 80a2156
1 parent 8e46fce
commit 80a2156
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 7 deletions.
diff --git a/factgenie/run.py → factgenie/bin/run.py b/factgenie/run.py → factgenie/bin/run.py
diff --git a/factgenie/datasets/hf_dataset.py b/factgenie/datasets/hf_dataset.py
@@ -1,6 +1,7 @@
 #!/usr/bin/env python3
 import logging
 import os
+from datasets import load_dataset
 
 from factgenie.datasets.dataset import Dataset
 
@@ -10,8 +11,6 @@
 class HFDataset(Dataset):
     @classmethod
     def download(cls, dataset_id, data_download_dir, out_download_dir, splits, outputs, dataset_config, **kwargs):
-        from datasets import load_dataset
-
         hf_id = kwargs.get("hf_id", dataset_id)
         out_column = kwargs.get("out_column", None)
 

diff --git a/setup.py b/setup.py
@@ -62,13 +62,13 @@ def run(self):
     include_package_data=True,
     entry_points={
         "console_scripts": [
-            "factgenie=factgenie.run:run",
+            "factgenie=factgenie.bin.run:run",
         ],
         "flask.commands": [
-            "create_llm_campaign=factgenie.run:create_llm_campaign",
-            "run_llm_campaign=factgenie.run:run_llm_campaign",
-            "list=factgenie.run:list_data",
-            "info=factgenie.run:info",
+            "create_llm_campaign=factgenie.bin.run:create_llm_campaign",
+            "run_llm_campaign=factgenie.bin.run:run_llm_campaign",
+            "list=factgenie.bin.run:list_data",
+            "info=factgenie.bin.run:info",
         ],
     },
     install_requires=install_requires,