From 5b1d7f72c70e82b1a9f9e3043a725579fdd6c492 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Fran=C3=A7ois=20Laurent?= <francois.laurent@posteo.net>
Date: Tue, 4 Apr 2023 13:11:38 +0200
Subject: [PATCH] hopefully fixes #1
---
pyproject.toml | 4 ++--
src/behavior_model/data/larva_dataset.py | 2 +-
2 files changed, 3 insertions(+), 3 deletions(-)
diff --git a/pyproject.toml b/pyproject.toml
index bed9c7c..0320fd5 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -12,10 +12,10 @@ python = "^3.8,<3.11"
h5py = "3.1.0"
pandas = "1.3.3"
PyYAML = "^6.0"
-torchvision = "^0.12.0"
+torchvision = "0.14.1"
attrdict = "^2.0.1"
tqdm = "^4.64.0"
-torch = "^1.11.0"
+torch = "1.13.1"
[build-system]
requires = ["poetry-core>=1.0.0"]
diff --git a/src/behavior_model/data/larva_dataset.py b/src/behavior_model/data/larva_dataset.py
index 1140518..ae35bd3 100644
--- a/src/behavior_model/data/larva_dataset.py
+++ b/src/behavior_model/data/larva_dataset.py
@@ -81,7 +81,7 @@ class LarvaDataset(Dataset):
def import_label(s):
s = s.decode('utf-8') if isinstance(s, bytes) else s
# MaggotUBA only understands BACK, BEND, HUNCH, ROLL, RUN, STOP.
- # With the `small_motion` branch, it only understands SMALL_MOTION.
+ # With the `small_motion` branch, it also understands SMALL_MOTION.
if s == 'CAST':
return 'BEND'
elif s == 'STOP_LARGE':
--
GitLab