download to local dir (#12)

Author: Borda

_requirements/test.txt

@@ -3,3 +3,5 @@ pytest >=6.0
 pytest-cov
 pytest-mock
 mypy ==1.13.0
+
+pytorch-lightning >=2.0

src/litmodels/cloud_io.py

@@ -73,7 +73,7 @@ def upload_model(
 
 def download_model(
     name: str,
-    download_dir: Optional[str] = None,
+    download_dir: str = ".",
     progress_bar: bool = True,
 ) -> str:
     """Download a checkpoint from the model store.