Merge branch 'release/0.1.2'

Author: cappelletto

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "bnn_inference"
-version = "0.1.1"
+version = "0.1.2"
 description = "Bayesian NN training/inference engine to learn mappings between latent representations of low-resolution maps and high-resolution observations"
 license = { file = "LICENSE" }
 readme = "README.md"
@@ -25,11 +25,11 @@ python = ">=3.10"
 blitz-bayesian-pytorch = "0.2.7"
 numpy = ">=1.19.0"
 jinja = ">=3.1.2"
-pandas = ">=0.25.3"
+pandas = ">=1.4.3"
 torch = ">=2.0.0"
-networkx = ">=3.1.0"
 torchvision = ">=0.15.1"
-scikit-learn = ">=0.23.1"
+networkx = ">=3.1.0"
+scikit-learn = ">=1.2.2"
 Pillow = ">=9.1.1"
 requests = "2.31.0"
 scipy = ">=1.5.0"

src/bnn_inference/train.py

@@ -227,7 +227,7 @@ def train_impl(
     )
 
     X_train, X_valid, y_train, y_valid = train_test_split(
-        X_norm, y_norm, test_size=xratio, shuffle=True  # 8:2 ratio
+        X_norm, y_norm, train_size=xratio, shuffle=True  # 8:2 ratio
     )
     # Convert train and test vectors to tensors
     X_train, y_train = torch.Tensor(X_train).float(), torch.Tensor(y_train).float()