tests: update path references

tieandrews · tieandrews · commit 1c5d66dc0c75 · 2023-06-27T17:48:52.000-07:00
diff --git a/tests/entity_extraction/test_baseline_entity_extraction.py b/tests/entity_extraction/test_baseline_entity_extraction.py
@@ -9,7 +9,7 @@
 # ensure that the parent directory is on the path for relative imports
 sys.path.append(os.path.join(os.path.dirname(__file__), "..", ".."))
 
-from src.entity_extraction.baseline_entity_extraction import (
+from src.entity_extraction.prediction.baseline_entity_extraction import (
     extract_geographic_coordinates,
     extract_region_names,
     extract_taxa,
diff --git a/tests/entity_extraction/test_entity_extraction_evaluation.py b/tests/entity_extraction/test_entity_extraction_evaluation.py
@@ -9,7 +9,7 @@
 # ensure that the parent directory is on the path for relative imports
 sys.path.append(os.path.join(os.path.dirname(__file__), "..", ".."))
 
-from src.entity_extraction.entity_extraction_evaluation import (
+from src.entity_extraction.evaluation.entity_extraction_evaluation import (
     get_token_labels,
     calculate_entity_classification_metrics,
     plot_token_classification_report,
diff --git a/tests/entity_extraction/test_hf_entity_extraction.py b/tests/entity_extraction/test_hf_entity_extraction.py
@@ -10,7 +10,7 @@
 # ensure that the parent directory is on the path for relative imports
 sys.path.append(os.path.join(os.path.dirname(__file__), "..", ".."))
 
-from src.entity_extraction.hf_entity_extraction import (
+from src.entity_extraction.prediction.hf_entity_extraction import (
     load_ner_model_pipeline,
     get_hf_token_labels,
     get_predicted_labels,
diff --git a/tests/entity_extraction/test_hf_evaluate.py b/tests/entity_extraction/test_hf_evaluate.py
@@ -10,7 +10,7 @@
 # ensure that the parent directory is on the path for relative imports
 sys.path.append(os.path.join(os.path.dirname(__file__), "..", ".."))
 
-from src.entity_extraction.training.hf_token_classification.hf_evaluate import (
+from src.entity_extraction.evaluation.hf_evaluate import (
     get_hf_token_labels,
     get_predicted_labels,
     generate_classification_results,
diff --git a/tests/entity_extraction/test_huggingface_preprocessing.py b/tests/entity_extraction/test_huggingface_preprocessing.py
@@ -9,7 +9,7 @@
 # ensure that the parent directory is on the path for relative imports
 sys.path.append(os.path.join(os.path.dirname(__file__), "..", ".."))
 
-from src.entity_extraction.training.hf_token_classification.huggingface_preprocess import (
+from src.entity_extraction.preprocessing.huggingface_preprocess import (
     convert_labelled_data_to_hf_format,
 )
 
diff --git a/tests/entity_extraction/test_labelling_data_split.py b/tests/entity_extraction/test_labelling_data_split.py
@@ -12,7 +12,7 @@
 # ensure that the parent directory is on the path for relative imports
 sys.path.append(os.path.join(os.path.dirname(__file__), "..", ".."))
 
-from src.preprocessing.labelling_data_split import separate_labels_to_train_val_test
+from src.entity_extraction.preprocessing.labelling_data_split import separate_labels_to_train_val_test
 
 
 # testing setup inspiration from: https://stackoverflow.com/questions/29627341/pytest-where-to-store-expected-data

Original file line number	Diff line number	Diff line change
`@@ -9,7 +9,7 @@`
`9`	`9`	`# ensure that the parent directory is on the path for relative imports`
`10`	`10`	`sys.path.append(os.path.join(os.path.dirname(__file__), "..", ".."))`
`11`	`11`
`12`		`-from src.entity_extraction.training.hf_token_classification.huggingface_preprocess import (`
	`12`	`+from src.entity_extraction.preprocessing.huggingface_preprocess import (`
`13`	`13`	`convert_labelled_data_to_hf_format,`
`14`	`14`	`)`
`15`	`15`