File tree Expand file tree Collapse file tree 2 files changed +0
-5
lines changed
jax_tpu_embedding/sparsecore
examples/models/shakespeare Expand file tree Collapse file tree 2 files changed +0
-5
lines changed Original file line number Diff line number Diff line change @@ -24,7 +24,6 @@ pytype_strict_library(
2424 srcs = [
2525 "dataset.py" ,
2626 ],
27- srcs_version = "PY3" ,
2827 deps = [
2928 pypi_requirement ("absl/logging" ),
3029 pypi_requirement ("numpy" ),
@@ -36,7 +35,6 @@ pytype_strict_library(
3635 srcs = [
3736 "model.py" ,
3837 ],
39- srcs_version = "PY3" ,
4038 deps = [
4139 "//jax_tpu_embedding/sparsecore/lib/nn:embedding_spec" ,
4240 pypi_requirement ("flax:core" ),
@@ -50,7 +48,6 @@ pytype_strict_library(
5048 srcs = [
5149 "flax_model.py" ,
5250 ],
53- srcs_version = "PY3" ,
5451 deps = [
5552 "//jax_tpu_embedding/sparsecore/lib/flax:embed" ,
5653 "//jax_tpu_embedding/sparsecore/lib/nn:embedding" ,
Original file line number Diff line number Diff line change @@ -93,7 +93,6 @@ pytype_strict_library(
9393 srcs = [
9494 "input_preprocessing.py" ,
9595 ],
96- srcs_version = "PY3" ,
9796 deps = [
9897 ":constants" ,
9998 pypi_requirement ("absl:app" ),
@@ -109,7 +108,6 @@ pytype_strict_library(
109108 srcs = [
110109 "constants.py" ,
111110 ],
112- srcs_version = "PY3" ,
113111)
114112
115113pytype_strict_contrib_test (
You can’t perform that action at this time.
0 commit comments