init__

Author: Borda

src/litmodels/__init__.py

@@ -7,6 +7,6 @@
 _PACKAGE_ROOT = os.path.dirname(__file__)
 _PROJECT_ROOT = os.path.dirname(_PACKAGE_ROOT)
 
-from litmodels.cloud_io import download_model_files, upload_model_files
+from litmodels.cloud_io import download_model_files, upload_model, upload_model_files
 
-__all__ = ["download_model_files", "upload_model_files"]
+__all__ = ["download_model_files", "upload_model", "upload_model_files"]