Upload folder using huggingface_hub

src/app.py

@@ -9,15 +9,15 @@ from fastapi.responses import JSONResponse, RedirectResponse, StreamingResponse
 from gymnasium.wrappers.record_video import RecordVideo
 from loguru import logger
 from pydantic import BaseModel
-from src.app_state import AppState
-from src.huggingface.huggingface_client import HuggingFaceClient
-from src.typing import CpuConfig
 
 from litrl.algo.mcts.agent import MCTSAgent
 from litrl.common.agent import RandomAgent
 from litrl.env.connect_four import ConnectFour
 from litrl.env.make import make
 from litrl.env.typing import GymId
+from src.app_state import AppState
+from src.huggingface.huggingface_client import HuggingFaceClient
+from src.typing import CpuConfig
 
 
 def stream_mp4(mp4_path: Path) -> StreamingResponse:

src/app_state.py

@@ -1,7 +1,6 @@
 from typing import Any, Self
 
 from loguru import logger
-from src.typing import AgentType, CpuConfig, RolloutPolicy
 
 from litrl import make_multiagent
 from litrl.algo.mcts.agent import MCTSAgent
@@ -10,6 +9,7 @@ from litrl.algo.mcts.rollout import VanillaRollout
 from litrl.common.agent import Agent, RandomMultiAgent
 from litrl.env.connect_four import ConnectFour
 from litrl.model.sac.multi_agent import OnnxSacDeterministicMultiAgent
+from src.typing import AgentType, CpuConfig, RolloutPolicy
 
 
 class AppState:

src/huggingface/get_environments.py

@@ -1,4 +1,5 @@
 from huggingface_hub import HfApi
+
 from src.constants import ENV_RESULTS_FILE_DEPTH, MODEL_REPO, MODEL_REPO_TYPE
 
 

src/huggingface/get_files.py

@@ -2,6 +2,7 @@ from pathlib import Path
 
 import yaml
 from huggingface_hub import hf_hub_download
+
 from src.constants import MODEL_REPO, MODEL_REPO_TYPE