Fix errors due to dev merge

Signed-off-by: Partho Sarthi <[email protected]>
NVIDIA · Sep 21, 2023 · 93d126b · 93d126b
1 parent 96f5e11
commit 93d126b
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 1 deletion.
diff --git a/...ests/spark_rapids_tools_ut/resources/cluster/databricks/test-azure-instances-catalog.json b/...ests/spark_rapids_tools_ut/resources/cluster/databricks/test-azure-instances-catalog.json
@@ -0,0 +1,22 @@
+{
+  "Standard_NC4as_T4_v3": {
+    "VCpuInfo": {
+      "DefaultVCpus": 4
+    },
+    "MemoryInfo": {
+      "SizeInMiB": 0
+    },
+    "GpuInfo": {
+      "GPUs": [
+        {
+          "Name": "",
+          "Manufacturer": "",
+          "Count": 1,
+          "MemoryInfo": {
+            "SizeInMiB": 0
+          }
+        }
+      ]
+    }
+  }
+}
diff --git a/user_tools/tests/test_diagnostic.py b/user_tools/tests/test_diagnostic.py
@@ -22,7 +22,7 @@
 from cli_test_helpers import ArgvContext, EnvironContext  # pylint: disable=import-error
 
 from spark_rapids_pytools import wrapper
-from .pyrapids_unit import conftest
+from .spark_rapids_tools_ut import conftest
 from .mock_cluster import mock_live_cluster