(#8) resolve issue with dependencies

SaintAngeLs · Jan 14, 2025 · 03dc43c · 03dc43c
1 parent 8cfa1c2
commit 03dc43c
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 61 deletions.
diff --git a/setup.py b/setup.py
@@ -11,7 +11,7 @@ def get_version():
 
 def increment_version(version):
     major, minor, patch = map(int, version.split("."))
-    patch += 1
+    patch += 2
     return f"{major}.{minor}.{patch}"
 
 current_version = get_version()

diff --git a/src/FeatureFlex/feature_selector.py b/src/FeatureFlex/feature_selector.py
@@ -2,7 +2,7 @@
 import torch.nn as nn
 import numpy as np
 import shap
-from model_optimizer import ModelOptimizer
+from .model_optimizer import ModelOptimizer
 from sklearn.ensemble import RandomForestClassifier