Some internal changes.

PiperOrigin-RevId: 714502148
google · Jan 18, 2025 · 0947135 · 0947135
1 parent 2a72d26
commit 0947135
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 14 deletions.
diff --git a/export/orbax/export/export_manager_obm_test.py b/export/orbax/export/export_manager_obm_test.py
@@ -37,13 +37,6 @@
 )
 
 
-def _get_expected_tf_function_body(name: str, support_tf_resources: bool):
-  if support_tf_resources:
-    return f'inlined_string: "{name}"'
-  else:
-    return 'file_system_location { string_path: "%s.pb" }' % name
-
-
 class ExportManagerObmTest(parameterized.TestCase, tf.test.TestCase):
 
   @parameterized.named_parameters(

diff --git a/export/orbax/export/obm_export_test.py b/export/orbax/export/obm_export_test.py
@@ -38,13 +38,6 @@ def _package_jax_module(m: jax_module.JaxModule):
   return result
 
 
-def _get_expected_tf_function_body(name: str, support_tf_resources: bool):
-  if support_tf_resources:
-    return f'inlined_string: "{name}"'
-  else:
-    return 'file_system_location { string_path: "%s.pb" }' % name
-
-
 class ObmExportTest(parameterized.TestCase, tf.test.TestCase):
 
   def setUp(self):