From 93d126b1797680477e5704a60896acf856c5c882 Mon Sep 17 00:00:00 2001 From: Partho Sarthi Date: Thu, 21 Sep 2023 11:25:45 -0700 Subject: [PATCH] Fix errors due to dev merge Signed-off-by: Partho Sarthi --- .../test-azure-instances-catalog.json | 22 +++++++++++++++++++ user_tools/tests/test_diagnostic.py | 2 +- 2 files changed, 23 insertions(+), 1 deletion(-) diff --git a/user_tools/tests/spark_rapids_tools_ut/resources/cluster/databricks/test-azure-instances-catalog.json b/user_tools/tests/spark_rapids_tools_ut/resources/cluster/databricks/test-azure-instances-catalog.json index e69de29bb..85080e8e8 100644 --- a/user_tools/tests/spark_rapids_tools_ut/resources/cluster/databricks/test-azure-instances-catalog.json +++ b/user_tools/tests/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 + } + } + ] + } + } +} \ No newline at end of file diff --git a/user_tools/tests/test_diagnostic.py b/user_tools/tests/test_diagnostic.py index be9326306..d15e17615 100644 --- 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