Merge branch 'main' into fco/reproducibility_study

Author: fcogidi

.github/dependabot.yml

@@ -0,0 +1,11 @@
+# To get started with Dependabot version updates, you'll need to specify which
+# package ecosystems to update and where the package manifests are located.
+# Please see the documentation for all configuration options:
+# https://docs.github.com/code-security/dependabot/dependabot-version-updates/configuration-options-for-the-dependabot.yml-file
+
+version: 2
+updates:
+  - package-ecosystem: "github-actions" # See documentation for possible values
+    directory: "/" # Location of package manifests
+    schedule:
+      interval: "weekly"

.github/workflows/code_checks.yml

@@ -26,13 +26,13 @@ jobs:
   run-code-check:
     runs-on: ubuntu-latest
     steps:
-      - uses: actions/checkout@v4.2.1
+      - uses: actions/checkout@v4.2.2
       - name: Install and configure Poetry
         uses: snok/install-poetry@v1
         with:
           virtualenvs-create: true
           virtualenvs-in-project: true
-      - uses: actions/setup-python@v5.2.0
+      - uses: actions/setup-python@v5.3.0
         with:
           python-version: '3.10'
           cache: 'poetry'

.github/workflows/integration_tests.yml

@@ -32,13 +32,13 @@ jobs:
   integration-tests:
     runs-on: ubuntu-latest
     steps:
-      - uses: actions/checkout@v4.2.1
+      - uses: actions/checkout@v4.2.2
       - name: Install and configure Poetry
         uses: snok/install-poetry@v1
         with:
           virtualenvs-create: true
           virtualenvs-in-project: true
-      - uses: actions/setup-python@v5.2.0
+      - uses: actions/setup-python@v5.3.0
         with:
           python-version: '3.10'
           cache: 'poetry'

.github/workflows/publish.yml

@@ -18,10 +18,10 @@ jobs:
         run: |
           sudo apt-get update
           sudo apt-get install libcurl4-openssl-dev libssl-dev
-      - uses: actions/checkout@v4.2.1
+      - uses: actions/checkout@v4.2.2
       - name: Install poetry
         run: python3 -m pip install --upgrade pip && python3 -m pip install poetry
-      - uses: actions/setup-python@v5.2.0
+      - uses: actions/setup-python@v5.3.0
         with:
           python-version: '3.10'
       - name: Build package

.pre-commit-config.yaml

@@ -16,13 +16,13 @@ repos:
       - id: check-toml
 
   - repo: https://github.com/python-poetry/poetry
-    rev: 1.8.3
+    rev: 1.8.4
     hooks:
       - id: poetry-check
         args: [--lock]
 
   - repo: https://github.com/astral-sh/ruff-pre-commit
-    rev: v0.6.9
+    rev: v0.7.2
     hooks:
       - id: ruff
         args: [--fix, --exit-non-zero-on-fix]
@@ -31,13 +31,13 @@ repos:
         types_or: [ python, pyi, jupyter ]
 
   - repo: https://github.com/crate-ci/typos
-    rev: v1.26.0
+    rev: v1.27.0
     hooks:
       - id: typos
         args: []
 
   - repo: https://github.com/pre-commit/mirrors-mypy
-    rev: v1.11.2
+    rev: v1.13.0
     hooks:
       - id: mypy
         entry: mypy

mmlearn/modules/metrics/retrieval_recall.py

@@ -100,7 +100,7 @@ def _is_distributed(self) -> bool:
         if self.distributed_available_fn is not None:
             distributed_available = self.distributed_available_fn
 
-        return distributed_available() if callable(distributed_available) else False  # type: ignore[no-any-return]
+        return distributed_available() if callable(distributed_available) else False
 
     def update(self, x: torch.Tensor, y: torch.Tensor, indexes: torch.Tensor) -> None:
         """Check shape, convert dtypes and add to accumulators.

poetry.lock

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "mmlearn"
-version = "0.1.0a0.dev7" # https://www.python.org/dev/peps/pep-0440/#version-schemes
+version = "0.1.0a0.dev8" # https://www.python.org/dev/peps/pep-0440/#version-schemes
 description = "A modular framework for research on multimodal representation learning."
 readme = "README.md"
 authors = ["Vector AI Engineering <ai_engineering@vectorinstitute.ai>"]
@@ -17,21 +17,21 @@ pyarrow = "^17.0.0"
 hydra-core = "^1.3.2"
 hydra-zen = "^0.13.0"
 hydra-submitit-launcher = "^1.2.0"
-transformers = "^4.45.0"
+transformers = "^4.46.1"
 torch = "^2.4.1"
 lightning = "^2.4.0"
 pandas = {version="^2.2.3", extras=["performance"]}
+bottleneck = "^1.4.2"
 torchvision = "^0.19.1"
 
 opencv-python = {version = "^4.10.0.84", optional = true}
-timm = {version = "^1.0.8", optional = true}
+timm = {version = "^1.0.11", optional = true}
 torchaudio = {version = "^2.4.1", optional = true}
-peft = {version = "^0.13.1", optional = true}
-
+peft = {version = "^0.13.2", optional = true}
 
 [tool.poetry.group.vision.dependencies]
 opencv-python = "^4.10.0.84"
-timm = "^1.0.9"
+timm = "^1.0.11"
 
 [tool.poetry.group.audio.dependencies]
 torchaudio = "^2.4.1"
@@ -40,33 +40,33 @@ torchaudio = "^2.4.1"
 optional = true
 
 [tool.poetry.group.peft.dependencies]
-peft = "^0.13.1"
+peft = "^0.13.2"
 
 [tool.poetry.group.dev]
 optional = true
 
 [tool.poetry.group.dev.dependencies]
-wandb = "^0.18.3"
+wandb = "^0.18.5"
 ipykernel = "^6.29.5"
-ipython = "8.28.0"
+ipython = "8.29.0"
 ipywidgets = "^8.1.5"
 h5py = "^3.12.1"
 scikit-learn = "^1.5.2"
 faiss-gpu = "^1.7.2"
-pip = "^24.2"
-datasets = "^3.0.1"
+pip = "^24.3.1"
+datasets = "^3.1.0"
 
 [tool.poetry.group.test]
 optional = true
 
 [tool.poetry.group.test.dependencies]
-ruff = "^0.6.9"
-mypy = "^1.11.2"
+ruff = "^0.7.2"
+mypy = "^1.13.0"
 pytest = "^8.3.3"
 pre-commit = "^4.0.1"
 nbqa = { version = "^1.9.0", extras = ["toolchain"] }
 pip-audit = "^2.7.3"
-typos = "^1.26.0"
+typos = "^1.27.0"
 
 [tool.poetry.extras]
 peft = ["peft"]