Upload folder using huggingface_hub

src/huggingface/get_files.py

@@ -23,7 +23,7 @@ def get_best(env: str, filename: str = "demo.mp4") -> Path:
         filename=hf_env_results_path,
     )
     with Path(local_env_results_path).open() as f:
-        env_results = yaml.load(f, Loader=yaml.FullLoader)
+        env_results = yaml.safe_load(f)
     best_model_type = max(env_results, key=lambda model: env_results[model])
     model_results_path = f"models/{env}/{best_model_type}/results.yaml"
     local_model_results_path = hf_hub_download(
@@ -32,7 +32,7 @@ def get_best(env: str, filename: str = "demo.mp4") -> Path:
         filename=model_results_path,
     )
     with Path(local_model_results_path).open() as f:
-        model_results = yaml.load(f, Loader=yaml.FullLoader)
+        model_results = yaml.safe_load(f)
 
     best_model = max(model_results, key=lambda model: model_results[model])
     hf_gif_path = f"models/{env}/{best_model_type}/{best_model}/{filename}"