added audio_dataset and audio_dataset_test to the BUILD file
@@ -286,6 +286,19 @@ py_library(
],
)
+
+py_library(
+ name = "audio_dataset",
+ srcs = [
+ "audio_dataset.py",
+ ],
+ srcs_version = "PY3",
+ deps = [
+ "//:expect_numpy_installed",
+ "//:expect_tensorflow_installed",
+ ],
+)
+
tf_py_test(
name = "dataset_creator_test",
srcs = ["dataset_creator_test.py"],
@@ -557,3 +570,17 @@ tf_py_test(
"//keras/testing_infra:test_utils",
],
)
+
+tf_py_test(
+ name = "audio_dataset_test",
+ size = "small",
+ srcs = ["audio_dataset_test.py"],
+ python_version = "PY3",
+ deps = [
+ ":audio_dataset",
+ "//:expect_numpy_installed",
+ "//:expect_tensorflow_installed",
+ "//keras/testing_infra:test_combinations",
+ "//keras/testing_infra:test_utils",
+ ],
+)
