fix stray code from chilc PRs

openxla · Jul 19, 2024 · b559dbe · b559dbe
1 parent 31dca25
commit b559dbe
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 11 deletions.
diff --git a/stablehlo/integrations/python/CMakeLists.txt b/stablehlo/integrations/python/CMakeLists.txt
@@ -65,15 +65,6 @@ declare_mlir_python_sources(StablehloToSavedModelPythonSources
     stablehlo/savedmodel/stablehlo_to_tf_saved_model.py
 )
 
-declare_mlir_python_sources(StablehloTestdataGeneratorPythonSources
-  ADD_TO_PARENT StablehloPythonSources
-  ROOT_DIR "${CMAKE_CURRENT_SOURCE_DIR}"
-  SOURCES
-    stablehlo/testdata_generator/testdata_execution_utils.py
-    stablehlo/testdata_generator/testdata_generator_lib.py
-    stablehlo/testdata_generator/testdata_processor.py
-)
-
 declare_mlir_python_sources(VhloPythonSources)
 declare_mlir_python_sources(VhloPythonSources.Dialects
   ADD_TO_PARENT VhloPythonSources

diff --git a/stablehlo/integrations/python/tests/check.py b/stablehlo/integrations/python/tests/check.py
@@ -23,8 +23,8 @@
 
 def run(f):
   with ir.Context() as context:
-    check.register_dialect(context)
-    stablehlo_dialect.register_dialect(ctx)
+    check_dialect.register_dialect(context)
+    stablehlo_dialect.register_dialect(context)
     f()
   return f