build(poetry): use datasets 1.11.0 to avoid mapping issues

formermagic · Sep 17, 2021 · 42c701f · 42c701f
1 parent 23f020e
commit 42c701f
Show file tree

Hide file tree

Showing 2 changed files with 79 additions and 60 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -22,7 +22,7 @@ flax = "^0.3.4"
 optax = "^0.0.9"
 torch = "^1.9.0"
 transformers = "^4.10.0"
-datasets = "^1.11.0"
+datasets = ">=1.11.0,<1.12.0"
 hydra-core = "^1.1.1"
 more-itertools = "^8.8.0"
 wandb = "^0.12.1"