bump up to 0.0.12

rsinghlab · Jan 9, 2024 · f7a5f3a · f7a5f3a
1 parent 54dd579
commit f7a5f3a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/pyaging/predict/_pred_utils.py b/pyaging/predict/_pred_utils.py
@@ -626,7 +626,7 @@ def predict_ages_with_model(
         for batch in main_tqdm(
             dataloader, indent_level=indent_level + 1, logger=logger
         ):
-            batch_pred = model(batch[:, features].X)
+            batch_pred = model(batch[:, features].X.to(device))
             predictions.append(batch_pred)
 
     # Concatenate all batch predictions

diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "pyaging"
-version = "0.0.11"
+version = "0.0.12"
 description = "A Python-based compendium of GPU-optimized aging clocks."
 authors = ["Lucas Paulo de Lima Camillo <[email protected]>"]
 license = "BSD"