Upgrade the required python version to python>=3.11

.github/workflows/docs-latest.yml

@@ -16,7 +16,7 @@ jobs:
  - name: Set up Python and uv
  uses: drivendataorg/setup-python-uv-action@v1
  with:
- python-version: 3.8
+ python-version: 3.11
 
  - name: Install dependencies
  run: |

.github/workflows/docs-preview.yml

@@ -14,7 +14,7 @@ jobs:
  - name: Set up Python and uv
  uses: drivendataorg/setup-python-uv-action@v1
  with:
- python-version: 3.8
+ python-version: 3.11
 
  - name: Install dependencies
  run: |

.github/workflows/release.yml

@@ -29,7 +29,7 @@ jobs:
  - name: Set up Python and uv
  uses: drivendataorg/setup-python-uv-action@v1
  with:
- python-version: 3.8
+ python-version: 3.11
 
  - name: Install dependencies
  run: |

.github/workflows/tests.yml

@@ -22,7 +22,7 @@ jobs:
  - name: Set up Python and uv
  uses: drivendataorg/setup-python-uv-action@v1
  with:
- python-version: 3.8
+ python-version: [3.11]
 
  - name: Install dependencies
  run: |
@@ -41,7 +41,7 @@ jobs:
  strategy:
  matrix:
  os: [ubuntu-latest, macos-latest]
- python-version: [3.8, 3.9]
+ python-version: [3.11, 3.12]
 
  steps:
  - if: matrix.os == 'ubuntu-latest'
@@ -119,7 +119,7 @@ jobs:
  strategy:
  matrix:
  os: [ubuntu-latest]
- python-version: [3.8]
+ python-version: [3.11]
 
  steps:
  - name: Remove unused software

Makefile

@@ -6,7 +6,7 @@
 
 PROJECT_DIR := $(shell dirname $(realpath $(lastword $(MAKEFILE_LIST))))
 PROJECT_NAME = zamba
-PYTHON_VERSION = 3.8
+PYTHON_VERSION = 3.11
 PYTHON_INTERPRETER = python
 
 ifeq (, $(shell which nvidia-smi))

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.11
 * FFmpeg > 4.3
 
 Then run:

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.11
 * FFmpeg > 4.3
 
 Then run:

docs/docs/install.md

@@ -9,10 +9,10 @@ GPU configurations.
 
 Prerequisites:
 
- - Python 3.8 or 3.9
+ - Python >= 3.11
  - FFmpeg
 
-#### [Python](https://www.python.org/) 3.8 or 3.9
+#### [Python](https://www.python.org/) >= 3.11
 
 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:
 

pyproject.toml

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

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