Remove srcs_version and python_version attributes, as they alread…

…y default to `"PY3"` PiperOrigin-RevId: 720986007
jax-ml · Jan 29, 2025 · b95d8bf · b95d8bf
1 parent 4ebefc2
commit b95d8bf
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 5 deletions.
diff --git a/jax_tpu_embedding/sparsecore/examples/models/shakespeare/BUILD b/jax_tpu_embedding/sparsecore/examples/models/shakespeare/BUILD
@@ -24,7 +24,6 @@ pytype_strict_library(
     srcs = [
         "dataset.py",
     ],
-    srcs_version = "PY3",
     deps = [
         pypi_requirement("absl/logging"),
         pypi_requirement("numpy"),
@@ -36,7 +35,6 @@ pytype_strict_library(
     srcs = [
         "model.py",
     ],
-    srcs_version = "PY3",
     deps = [
         "//jax_tpu_embedding/sparsecore/lib/nn:embedding_spec",
         pypi_requirement("flax:core"),
@@ -50,7 +48,6 @@ pytype_strict_library(
     srcs = [
         "flax_model.py",
     ],
-    srcs_version = "PY3",
     deps = [
         "//jax_tpu_embedding/sparsecore/lib/flax:embed",
         "//jax_tpu_embedding/sparsecore/lib/nn:embedding",

diff --git a/jax_tpu_embedding/sparsecore/lib/core/BUILD b/jax_tpu_embedding/sparsecore/lib/core/BUILD
@@ -93,7 +93,6 @@ pytype_strict_library(
     srcs = [
         "input_preprocessing.py",
     ],
-    srcs_version = "PY3",
     deps = [
         ":constants",
         pypi_requirement("absl:app"),
@@ -109,7 +108,6 @@ pytype_strict_library(
     srcs = [
         "constants.py",
     ],
-    srcs_version = "PY3",
 )
 
 pytype_strict_contrib_test(