Filter files found in /models

brabbit61 · brabbit61 · commit e6cadc8f72d1 · 2023-06-28T12:22:10.000-07:00
diff --git a/src/article_relevance/relevance_prediction_parquet.py b/src/article_relevance/relevance_prediction_parquet.py
@@ -426,7 +426,9 @@ def main():
     send_xdd = opt['--send_xdd']
     
     # /models directory is a mounted volume, containing the model object
-    models = os.lsitdir("/models")
+    models = os.listdir("/models")
+    models = [f for f in models if f.endswith(".joblib")]
+    
     if models:
         model_path = os.path.join("/models", models[0])
     else:
