diff --git a/tensorflow_io/core/filesystems/filesystem_plugins.cc b/tensorflow_io/core/filesystems/filesystem_plugins.cc index 1f0f6cab2..5e75e5a6f 100644 --- a/tensorflow_io/core/filesystems/filesystem_plugins.cc +++ b/tensorflow_io/core/filesystems/filesystem_plugins.cc @@ -29,9 +29,9 @@ limitations under the License. TFIO_PLUGIN_EXPORT void TF_InitPlugin(TF_FilesystemPluginInfo* info) { info->plugin_memory_allocate = tensorflow::io::plugin_memory_allocate; info->plugin_memory_free = tensorflow::io::plugin_memory_free; - info->num_schemes = 7; -#if !defined(_MSC_VER) info->num_schemes = 8; +#if !defined(_MSC_VER) + info->num_schemes = 9; #endif info->ops = static_cast( tensorflow::io::plugin_memory_allocate(info->num_schemes * @@ -43,7 +43,8 @@ TFIO_PLUGIN_EXPORT void TF_InitPlugin(TF_FilesystemPluginInfo* info) { tensorflow::io::hdfs::ProvideFilesystemSupportFor(&info->ops[4], "hdfs"); tensorflow::io::hdfs::ProvideFilesystemSupportFor(&info->ops[5], "viewfs"); tensorflow::io::hdfs::ProvideFilesystemSupportFor(&info->ops[6], "har"); + tensorflow::io::hdfs::ProvideFilesystemSupportFor(&info->ops[7], "qbfs"); #if !defined(_MSC_VER) - tensorflow::io::oss::ProvideFilesystemSupportFor(&info->ops[7], "oss"); + tensorflow::io::oss::ProvideFilesystemSupportFor(&info->ops[8], "oss"); #endif } diff --git a/tensorflow_io/core/filesystems/hdfs/hadoop_filesystem.cc b/tensorflow_io/core/filesystems/hdfs/hadoop_filesystem.cc index 3d5a41b84..68a88aece 100644 --- a/tensorflow_io/core/filesystems/hdfs/hadoop_filesystem.cc +++ b/tensorflow_io/core/filesystems/hdfs/hadoop_filesystem.cc @@ -541,6 +541,8 @@ hdfsFS Connect(tf_hdfs_filesystem::HadoopFileSystemImplementation* hadoop_file, std::string path_har = path; SplitArchiveNameAndPath(&path_har, &namenode, status); if (TF_GetCode(status) != TF_OK) return nullptr; + } else if (scheme == "qbfs") { + namenode = path; } else { if (namenode.empty()) { namenode = "default";