Skip to content

Commit a0a4dc7

Browse files
authored
Bump TF to 2.4.0rc3, and enable TF C API logging (#1206)
* Bump TF to 2.4.0rc3, and enable TF C API logging This PR bumps TF to 2.4.0rc3, and enable TF C API logging Signed-off-by: Yong Tang <yong.tang.github@outlook.com> * Add `bash -x -e tests/test_pulsar/pulsar_test.sh` in kokoro setup Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
1 parent 162da25 commit a0a4dc7

File tree

10 files changed

+12
-43
lines changed

10 files changed

+12
-43
lines changed

.kokorun/io_cpu.sh

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -73,6 +73,7 @@ ls wheelhouse/*
7373
## Set test services
7474
bash -x -e tests/test_gcloud/test_gcs.sh gcs-emulator
7575
bash -x -e tests/test_kafka/kafka_test.sh
76+
bash -x -e tests/test_pulsar/pulsar_test.sh
7677
bash -x -e tests/test_aws/aws_test.sh
7778
bash -x -e tests/test_pubsub/pubsub_test.sh pubsub
7879
bash -x -e tests/test_prometheus/prometheus_test.sh start

tensorflow_io/core/plugins/az/az_file_system.cc

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -31,11 +31,7 @@ limitations under the License.
3131
#include "storage_account.h"
3232
#include "storage_credential.h"
3333
#include "storage_errno.h"
34-
// TODO: Restore logging.h
35-
#define TF_Log(...)
36-
#define TF_VLog(...)
37-
// #include "tensorflow/c/logging.h"
38-
// TODO: Restore logging.h
34+
#include "tensorflow/c/logging.h"
3935
#include "tensorflow/c/tf_status.h"
4036
#include "tensorflow_io/core/plugins/file_system_plugins.h"
4137

tensorflow_io/core/plugins/gs/gcs_env.cc

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -28,12 +28,8 @@ limitations under the License.
2828
#endif
2929
#include <string>
3030
#include <vector>
31-
// TODO: Restore logging.h
32-
#define TF_Log(...)
33-
#define TF_VLog(...)
34-
// #include "tensorflow/c/logging.h"
35-
// TODO: Restore logging.h
3631

32+
#include "tensorflow/c/logging.h"
3733
#include "tensorflow_io/core/plugins/gs/gcs_env.h"
3834

3935
namespace tensorflow {

tensorflow_io/core/plugins/gs/gcs_filesystem.cc

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -20,16 +20,12 @@ limitations under the License.
2020
#include "absl/strings/str_cat.h"
2121
#include "absl/types/variant.h"
2222
#include "google/cloud/storage/client.h"
23+
#include "tensorflow/c/logging.h"
24+
#include "tensorflow/c/tf_status.h"
2325
#include "tensorflow_io/core/plugins/file_system_plugins.h"
2426
#include "tensorflow_io/core/plugins/gs/expiring_lru_cache.h"
2527
#include "tensorflow_io/core/plugins/gs/gcs_helper.h"
2628
#include "tensorflow_io/core/plugins/gs/ram_file_block_cache.h"
27-
// TODO: Restore logging.h
28-
#define TF_Log(...)
29-
#define TF_VLog(...)
30-
// #include "tensorflow/c/logging.h"
31-
// TODO: Restore logging.h
32-
#include "tensorflow/c/tf_status.h"
3329

3430
namespace tensorflow {
3531
namespace io {

tensorflow_io/core/plugins/gs/ram_file_block_cache.h

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -27,12 +27,8 @@ limitations under the License.
2727
#include "absl/base/thread_annotations.h"
2828
#include "absl/synchronization/mutex.h"
2929
#include "absl/synchronization/notification.h"
30+
#include "tensorflow/c/logging.h"
3031
#include "tensorflow/c/tf_status.h"
31-
// TODO: Restore logging.h
32-
#define TF_Log(...)
33-
#define TF_VLog(...)
34-
// #include "tensorflow/c/logging.h"
35-
// TODO: Restore logging.h
3632
#include "tensorflow_io/core/plugins/gs/gcs_env.h"
3733

3834
namespace tensorflow {

tensorflow_io/core/plugins/hdfs/hadoop_filesystem.cc

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -31,11 +31,7 @@ limitations under the License.
3131
#include "absl/strings/str_cat.h"
3232
#include "absl/synchronization/mutex.h"
3333
#include "hdfs/hdfs.h"
34-
// TODO: Restore logging.h
35-
#define TF_Log(...)
36-
#define TF_VLog(...)
37-
// #include "tensorflow/c/logging.h"
38-
// TODO: Restore logging.h
34+
#include "tensorflow/c/logging.h"
3935
#include "tensorflow/c/tf_status.h"
4036
#include "tensorflow_io/core/plugins/file_system_plugins.h"
4137

tensorflow_io/core/plugins/http/http_file_system.cc

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -22,11 +22,7 @@ limitations under the License.
2222
#include "absl/strings/ascii.h"
2323
#include "absl/strings/str_cat.h"
2424
#include "absl/synchronization/mutex.h"
25-
// TODO: Restore logging.h
26-
#define TF_Log(...)
27-
#define TF_VLog(...)
28-
// #include "tensorflow/c/logging.h"
29-
// TODO: Restore logging.h
25+
#include "tensorflow/c/logging.h"
3026
#include "tensorflow/c/tf_status.h"
3127
#include "tensorflow_io/core/plugins/file_system_plugins.h"
3228

tensorflow_io/core/plugins/s3/aws_logging.cc

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -23,11 +23,7 @@ limitations under the License.
2323
#include <sstream>
2424

2525
#include "absl/synchronization/mutex.h"
26-
// TODO: Restore logging.h
27-
#define TF_Log(...)
28-
#define TF_VLog(...)
29-
// #include "tensorflow/c/logging.h"
30-
// TODO: Restore logging.h
26+
#include "tensorflow/c/logging.h"
3127

3228
namespace tensorflow {
3329
namespace io {

tensorflow_io/core/plugins/s3/s3_filesystem.cc

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -36,15 +36,11 @@ limitations under the License.
3636
#include "absl/strings/ascii.h"
3737
#include "absl/strings/numbers.h"
3838
#include "absl/strings/str_cat.h"
39+
#include "tensorflow/c/logging.h"
40+
#include "tensorflow/c/tf_status.h"
3941
#include "tensorflow_io/core/plugins/file_system_plugins.h"
4042
#include "tensorflow_io/core/plugins/s3/aws_crypto.h"
4143
#include "tensorflow_io/core/plugins/s3/aws_logging.h"
42-
// TODO: Restore logging.h
43-
#define TF_Log(...)
44-
#define TF_VLog(...)
45-
// #include "tensorflow/c/logging.h"
46-
// TODO: Restore logging.h
47-
#include "tensorflow/c/tf_status.h"
4844

4945
namespace tensorflow {
5046
namespace io {

tensorflow_io/core/python/ops/version_ops.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,5 +14,5 @@
1414
# ==============================================================================
1515
"""version_ops"""
1616

17-
package = "tensorflow>=2.4.0rc2,<2.5.0"
17+
package = "tensorflow>=2.4.0rc3,<2.5.0"
1818
version = "0.17.0"

0 commit comments

Comments
 (0)