edit test to download encoder

Signed-off-by: Joozef315 <[email protected]>

.gitignore

@@ -94,4 +94,7 @@ dmypy.json
 # misc
 *.mp4
 sweep*/
-core*
+core*
+
+features_outputs
+*.pth

tests/feature_extraction_test.py

@@ -1,13 +1,15 @@
 import unittest
 from translation.feature_extraction_module import FeatureExtractionConfig, FeatureExtractionModule, LauncherConfig
 from unittest.mock import patch, MagicMock
+from utils.download_model import get_model_path
 
 class TestFeatureExtractionModule(unittest.TestCase):
     def setUp(self):
         # Mock the configuration for the FeatureExtractionModule
+        model_path = get_model_path()
         self.config = FeatureExtractionConfig(
             data_dir="MOCK_dataset",
-            pretrained_model_path="signhiera_mock.pth", 
+            pretrained_model_path=model_path, 
             launcher=LauncherConfig(cluster="local")
         )
         self.module = FeatureExtractionModule(self.config)

utils/download_model.py

@@ -0,0 +1,16 @@
+import os
+import wget
+
+def get_model_path():
+    model_path = 'signhiera_mock.pth'
+    url = 'https://dl.fbaipublicfiles.com/SONAR/asl/signhiera_mock.pth'
+
+    # Check if the model file exists
+    if os.path.exists(model_path):
+        print(f"Model already exists at: {model_path}")
+    else:
+        print("Model not found, downloading...")
+        filename = wget.download(url, model_path)
+        print(f"Downloaded model to: {filename}")
+
+    return model_path