upd dependency and fix FutureWarning

AIRI-Institute · Aug 4, 2024 · 04a3bf8 · 04a3bf8
1 parent 4c3e43a
commit 04a3bf8
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/fdd_defense/models/tcn.py b/fdd_defense/models/tcn.py
@@ -81,9 +81,9 @@ def __init__(
             dilation=(dilation_base**nr_blocks_below),
         )
         if weight_norm:
-            self.conv1, self.conv2 = nn.utils.weight_norm(
+            self.conv1, self.conv2 = nn.utils.parametrizations.weight_norm(
                 self.conv1
-            ), nn.utils.weight_norm(self.conv2)
+            ), nn.utils.parametrizations.weight_norm(self.conv2)
 
         if input_dim != output_dim:
             self.conv3 = nn.Conv1d(input_dim, output_dim, 1)

diff --git a/pyproject.toml b/pyproject.toml
@@ -11,7 +11,8 @@ pandas = "*"
 scikit-learn = "*"
 tqdm = "*"
 fddbenchmark = { git = "https://github.com/airi-industrial-ai/fddbenchmark.git" }
-catboost = "1.1.1"
+catboost = "1.2.5"
+numpy = "< 2.0"
 
 [build-system]
 requires = ["poetry-core>=1.0.0"]