From d6d85938301defd482c1005630a46496502a44b7 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Pawe=C5=82=20Redzy=C5=84ski?= Date: Mon, 14 Feb 2022 14:10:40 +0100 Subject: [PATCH] catalyst: restrict version --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index 10d1a87d..c1c51c26 100644 --- a/setup.py +++ b/setup.py @@ -43,7 +43,7 @@ def run(self): xgb = ["xgboost"] lgbm = ["lightgbm"] hugginface = ["transformers", "datasets"] -catalyst = ["catalyst"] +catalyst = ["catalyst<=21.12"] fastai = ["fastai"] pl = ["pytorch_lightning"]