Removed named_shape= from jax.core.ShapedArray and `jax.ShapeDtyp…

…eStruct` It is unused and was only kept around to avoid breaking internal users. PiperOrigin-RevId: 673411837
google-deepmind · Sep 11, 2024 · d07f5cb · d07f5cb
1 parent e78349f
commit d07f5cb
Showing 1 changed file with 0 additions and 6 deletions.
diff --git a/tests/deprecated/v1/toolshed/sharding_util_test.py b/tests/deprecated/v1/toolshed/sharding_util_test.py
@@ -200,7 +200,6 @@ def test_name_to_name_sharding(self):
                   data_array=jax.ShapeDtypeStruct(
                       shape=(4, 8, 6),
                       dtype=np.dtype('float32'),
-                      named_shape={},
                       sharding=jax.sharding.NamedSharding(
                           explicit_mesh, P('mesh_axis_1', 'mesh_axis_0', None)
                       ),
@@ -211,7 +210,6 @@ def test_name_to_name_sharding(self):
                   data_array=jax.ShapeDtypeStruct(
                       shape=(8,),
                       dtype=np.dtype('float32'),
-                      named_shape={},
                       sharding=jax.sharding.NamedSharding(
                           explicit_mesh, P('mesh_axis_1')
                       ),
@@ -222,7 +220,6 @@ def test_name_to_name_sharding(self):
                   data_array=jax.ShapeDtypeStruct(
                       shape=(8, 12),
                       dtype=np.dtype('float32'),
-                      named_shape={},
                       sharding=jax.sharding.NamedSharding(
                           explicit_mesh, P('mesh_axis_0', None)
                       ),
@@ -251,7 +248,6 @@ def test_name_to_name_sharding(self):
                   data_array=jax.ShapeDtypeStruct(
                       shape=(4, 8, 6),
                       dtype=np.dtype('float32'),
-                      named_shape={},
                       sharding=jax.sharding.NamedSharding(
                           explicit_mesh, P('mesh_axis_1', 'mesh_axis_0', None)
                       ),
@@ -264,7 +260,6 @@ def test_name_to_name_sharding(self):
                   data_array=jax.ShapeDtypeStruct(
                       shape=(8,),
                       dtype=np.dtype('float32'),
-                      named_shape={},
                       sharding=jax.sharding.NamedSharding(
                           explicit_mesh, P('mesh_axis_1')
                       ),
@@ -277,7 +272,6 @@ def test_name_to_name_sharding(self):
                   data_array=jax.ShapeDtypeStruct(
                       shape=(8, 12),
                       dtype=np.dtype('float32'),
-                      named_shape={},
                       sharding=jax.sharding.NamedSharding(
                           explicit_mesh, P('mesh_axis_0', None)
                       ),