address comments

Signed-off-by: Nitish Bharambe <[email protected]>

docs/examples/Power Flow Example.ipynb

@@ -402,7 +402,7 @@
       "List of attribute types in line result\n",
       "['id', 'p_from']\n",
       "------line result------\n",
-      "{'id': array([3, 5, 8]), 'p_from': array([17360100.20222374, -3365613.74450156, 13854413.52498137])}\n"
+      "{'id': array([3, 5, 8], dtype=int32), 'p_from': array([17360100.20222374, -3365613.74450156, 13854413.52498137])}\n"
      ]
     }
    ],
@@ -1406,11 +1406,11 @@
       "[[9.99401170e-001 9.92685785e-001 9.94521366e-001]\n",
       " [9.99347687e-001 9.86226389e-001 9.89352855e-001]\n",
       " [9.99288384e-001 9.79654011e-001 9.84095542e-001]\n",
-      " [6.95296562e-310 8.59448882e-312 6.95296562e-310]\n",
+      " [3.94357132e+180 2.87518198e+161 2.04418455e+214]\n",
       " [9.99151380e-001 9.66149483e-001 9.73298790e-001]\n",
       " [9.99073166e-001 9.59205860e-001 9.67750710e-001]\n",
       " [9.98988099e-001 9.52126208e-001 9.62096474e-001]\n",
-      " [8.59448882e-312 8.59448882e-312 6.95296562e-310]\n",
+      " [0.00000000e+000 0.00000000e+000 0.00000000e+000]\n",
       " [9.98796126e-001 9.37530046e-001 9.50447962e-001]\n",
       " [9.98688504e-001 9.29997471e-001 9.44441670e-001]]\n",
       "Node data with only valid results\n",
@@ -1469,7 +1469,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.12.7"
+   "version": "3.13.0"
   }
  },
  "nbformat": 4,

tests/unit/test_meta_data.py

@@ -82,6 +82,11 @@ def test_dict_like_access():
 
 def test_all_datasets():
     assert set(power_grid_meta_data.keys()) == {
+        "input",
+        "update",
+        "sym_output",
+        "asym_output",
+        "sc_output",
         DatasetType.input,
         DatasetType.update,
         DatasetType.sym_output,