Another try at dependency fixes

hannah/nas/performance_prediction/features/dataset.py

@@ -149,14 +149,13 @@ def get_features(nx_graph):
         df = unfold_columns(df, columns=get_list_columns(df))
         dataframes.append(df)
     df = pd.concat(dataframes)
-    # df.dropna(axis = 0, how = 'all', inplace = True)
     df = pd.get_dummies(df, dummy_na=True)
     df = df.fillna(0)
     for col in COLUMNS:
         if col not in df.columns:
             df[col] = 0
     df = df.reindex(sorted(df.columns), axis=1)  # Sort to have consistency
-    return df.astype(np.float32)
+    return df
 
 def get_list_columns(df):
     list_cols = []

pyproject.toml

@@ -62,19 +62,26 @@ onnx = "^1.16.0"
 spox = "^0.12.0"
 optree = "^0.11.0"
 pydantic = "^2.8.2"
-dgl = [
-  {platform = 'linux', python='~3.9', url = "https://data.dgl.ai/wheels/torch-2.3/cu121/dgl-2.3.0%2Bcu121-cp39-cp39-manylinux1_x86_64.whl"},
-  {platform = 'linux', python='~3.10', url = "https://data.dgl.ai/wheels/torch-2.3/cu121/dgl-2.3.0%2Bcu121-cp310-cp310-manylinux1_x86_64.whl"},
-  {platform = 'linux', python='~3.11', url = "https://data.dgl.ai/wheels/torch-2.3/cu121/dgl-2.3.0%2Bcu121-cp311-cp311-manylinux1_x86_64.whl"},
-  {platform = 'linux', python='~3.12', url = "https://data.dgl.ai/wheels/torch-2.3/cu121/dgl-2.3.0%2Bcu121-cp312-cp312-manylinux1_x86_64.whl"},
+#dgl = [
+#  {platform = 'linux', python='~3.9', url = "https://data.dgl.ai/wheels/torch-2.3/cu121/dgl-2.3.0%2Bcu121-cp39-cp39-manylinux1_x86_64.whl"},
+#  {platform = 'linux', python='~3.10', url = "https://data.dgl.ai/wheels/torch-2.3/cu121/dgl-2.3.0%2Bcu121-cp310-cp310-manylinux1_x86_64.whl"},
+#  {platform = 'linux', python='~3.11', url = "https://data.dgl.ai/wheels/torch-2.3/cu121/dgl-2.3.0%2Bcu121-cp311-cp311-manylinux1_x86_64.whl"},
+#  {platform = 'linux', python='~3.12', url = "https://data.dgl.ai/wheels/torch-2.3/cu121/dgl-2.3.0%2Bcu121-cp312-cp312-manylinux1_x86_64.whl"},
+#]
 
+dgl = [
+  {platform = 'linux', python='~3.9', url = "https://data.dgl.ai/wheels/torch-2.3/dgl-2.3.0-cp39-cp39-manylinux1_x86_64.whl"},
+  {platform = 'linux', python='~3.10', url = "https://data.dgl.ai/wheels/torch-2.3/dgl-2.3.0-cp310-cp310-manylinux1_x86_64.whl"},
+  {platform = 'linux', python='~3.11', url = "https://data.dgl.ai/wheels/torch-2.3/dgl-2.3.0-cp311-cp311-manylinux1_x86_64.whl"},
+  {platform = 'linux', python='~3.12', url = "https://data.dgl.ai/wheels/torch-2.3/dgl-2.3.0-cp312-cp312-manylinux1_x86_64.whl"},
 ]
 
 
 
 
 
 
+
 [tool.poetry.dev-dependencies]
 pytest = ">=7.2.0"
 pre_commit = ">=2.7.1"