fix bugs

yingyuankai · Feb 2, 2020 · bffd121 · bffd121
1 parent 98f1818
commit bffd121
Showing 1 changed file with 6 additions and 13 deletions.
diff --git a/tests/test_glue_zh.py b/tests/test_glue_zh.py
@@ -23,17 +23,10 @@ def test_glue_load(self):
         checksum_dir = "../aispace/datasets/url_checksums"
         tfds.download.add_checksums_dir(checksum_dir)
         # download_config = DownloadConfig(register_checksums=False)
-        tnews = tfds.load("glue_zh/tnews",
-                          # data_dir="/search/data1/yyk/data/datasets/glue_zh",
-                          data_dir="../data/glue_zh",
-                          builder_kwargs={'hparams': hparams},
-                          # download_and_prepare_kwargs={'download_config': download_config}
-                          )
-        for item in tnews['test']:
-            print()
-            # print(item["context"].numpy().decode("utf8"))
-            # print(item["question"].numpy().decode("utf8"))
-            print(item["sentence"].numpy().decode("utf8"))
-            print(item)
-            break
+        # tnews = tfds.load("glue_zh/tnews",
+        #                   # data_dir="/search/data1/yyk/data/datasets/glue_zh",
+        #                   data_dir="../data/glue_zh",
+        #                   builder_kwargs={'hparams': hparams},
+        #                   # download_and_prepare_kwargs={'download_config': download_config}
+        #                   )
         print()