Upgrade the required python version to python>=3.10

drivendataorg · Sep 5, 2024 · 4fc6f8a · 4fc6f8a
1 parent 498bca1
commit 4fc6f8a
Show file tree

Hide file tree

Showing 5 changed files with 8 additions and 8 deletions.
diff --git a/README.md b/README.md
@@ -29,7 +29,7 @@ Visit https://zamba.drivendata.org/docs/ for full documentation and tutorials.
 
 First, make sure you have the prerequisites installed:
 
-* Python 3.8 or 3.9
+* Python >= 3.10
 * FFmpeg > 4.3
 
 Then run:

diff --git a/docs/docs/index.md b/docs/docs/index.md
@@ -29,7 +29,7 @@ We encourage people to share their custom models trained with Zamba. If you trai
 
 First, make sure you have the prerequisites installed:
 
-* Python 3.8 or 3.9
+* Python >= 3.10
 * FFmpeg > 4.3
 
 Then run:

diff --git a/docs/docs/install.md b/docs/docs/install.md
@@ -9,10 +9,10 @@ GPU configurations.
 
 Prerequisites:
 
- - Python 3.8 or 3.9
+ - Python >= 3.10
  - FFmpeg
 
-#### [Python](https://www.python.org/) 3.8 or 3.9
+#### [Python](https://www.python.org/) >= 3.10
 
 We recommend [Python installation using Anaconda](https://www.anaconda.com/download/) for all platforms. For more information about how to install Anaconda, here are some useful YouTube videos of installation:
 

diff --git a/pyproject.toml b/pyproject.toml
@@ -16,7 +16,7 @@ classifiers = [
   "License :: OSI Approved :: MIT License",
   "Operating System :: OS Independent",
 ]
-requires-python = ">=3.8"
+requires-python = ">=3.10"
 dependencies = [
   "appdirs",
   "av",
@@ -38,7 +38,7 @@ dependencies = [
   "pytorchvideo",
   "scikit-learn",
   "tensorboard",
-  "thop==0.0.31.post2005241907",
+  "thop",
   "timm",
   "torch",
   "torchinfo",

diff --git a/zamba/models/config.py b/zamba/models/config.py
@@ -428,7 +428,7 @@ class TrainConfig(ZambaBaseModel):
     data_dir: DirectoryPath = ""
     checkpoint: Optional[FilePath] = None
     scheduler_config: Optional[Union[str, SchedulerConfig]] = "default"
-    model_name: Optional[ModelEnum] = ModelEnum.time_distributed
+    model_name: Optional[ModelEnum] = ModelEnum.time_distributed.value
     dry_run: Union[bool, int] = False
     batch_size: int = 2
     auto_lr_find: bool = False
@@ -749,7 +749,7 @@ class PredictConfig(ZambaBaseModel):
     data_dir: DirectoryPath = ""
     filepaths: Optional[FilePath] = None
     checkpoint: Optional[FilePath] = None
-    model_name: Optional[ModelEnum] = ModelEnum.time_distributed
+    model_name: Optional[ModelEnum] = ModelEnum.time_distributed.value
     gpus: int = GPUS_AVAILABLE
     num_workers: int = 3
     batch_size: int = 2