Fix imports

src/training/src/training/fine_tuning_cl/components.py

@@ -6,7 +6,7 @@
 from sklearn.model_selection import train_test_split
 from torch.utils.data import DataLoader, Dataset
 
-from amazon_product_search.training.shared.metric_logger import MetricLoggerST
+from training.shared.metric_logger import MetricLoggerST
 
 
 def query_to_text(query: str, with_tag: bool) -> str:

src/training/src/training/fine_tuning_cl/pipeline.py

@@ -30,7 +30,7 @@ def fine_tune(
 ) -> None:
     from collections import defaultdict
 
-    from amazon_product_search.training.fine_tuning_cl.components import run
+    from training.fine_tuning_cl.components import run
 
     metrics: list[dict[str, Any]] = run(
         project_dir,

src/training/src/training/fine_tuning_mlm/components.py

@@ -11,8 +11,8 @@
     AutoModelForMaskedLM,
 )
 
-from amazon_product_search.training.fine_tuning_mlm.data_module import ProductMLMDataModule
-from amazon_product_search.training.shared.metric_logger import MetricLoggerPL
+from training.fine_tuning_mlm.data_module import ProductMLMDataModule
+from training.shared.metric_logger import MetricLoggerPL
 
 
 class MLMFineTuner(LightningModule):

src/training/src/training/fine_tuning_mlm/pipeline.py

@@ -29,7 +29,7 @@ def fine_tune(
 ) -> None:
     from collections import defaultdict
 
-    from amazon_product_search.training.fine_tuning_mlm.components import run
+    from training.fine_tuning_mlm.components import run
 
     metrics: list[dict[str, Any]] = run(
         project_dir,