test: fixed TypeError checks

mwong009 · Mar 6, 2024 · 6269f13 · 6269f13
1 parent e12f00c
commit 6269f13
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_dataset.py b/tests/test_dataset.py
@@ -158,4 +158,4 @@ def test_make_tensor(lpmc_ds):
     assert tensors1.ndim == tensors2.ndim
 
     with pytest.raises(TypeError):
-        tensors3 = ds[[ds["cost_transit"], "cost_driving_ccharge"]]
+        tensors3 = ds[ds["cost_transit"], "cost_driving_ccharge", np.ones((3, 2))]