From ff1fd7ca41f9631118a37888ca59056aefe84748 Mon Sep 17 00:00:00 2001
From: Aleksey Veresov <aleksey@veresov.pro>
Date: Wed, 17 Jul 2024 18:31:13 +0200
Subject: [PATCH] Fix __all__ in client/__init__

---
 python/hopsworks/client/__init__.py           | 12 ++++++++
 .../client/online_store_rest_client.py        | 28 +++++++++++++++++++
 python/hsfs/client/__init__.py                | 12 ++++++++
 3 files changed, 52 insertions(+)
 create mode 100644 python/hopsworks/client/online_store_rest_client.py

diff --git a/python/hopsworks/client/__init__.py b/python/hopsworks/client/__init__.py
index 4ff5e8248..19e0feb8d 100644
--- a/python/hopsworks/client/__init__.py
+++ b/python/hopsworks/client/__init__.py
@@ -15,14 +15,26 @@
 #
 
 from hopsworks_common.client import (
+    auth,
+    base,
+    exceptions,
+    external,
     get_instance,
+    hopsworks,
     init,
+    online_store_rest_client,
     stop,
 )
 
 
 __all__ = [
+    auth,
+    base,
+    exceptions,
+    external,
     get_instance,
+    hopsworks,
     init,
+    online_store_rest_client,
     stop,
 ]
diff --git a/python/hopsworks/client/online_store_rest_client.py b/python/hopsworks/client/online_store_rest_client.py
new file mode 100644
index 000000000..c75be81b7
--- /dev/null
+++ b/python/hopsworks/client/online_store_rest_client.py
@@ -0,0 +1,28 @@
+#
+#   Copyright 2024 Hopsworks AB
+#
+#   Licensed under the Apache License, Version 2.0 (the "License");
+#   you may not use this file except in compliance with the License.
+#   You may obtain a copy of the License at
+#
+#       http://www.apache.org/licenses/LICENSE-2.0
+#
+#   Unless required by applicable law or agreed to in writing, software
+#   distributed under the License is distributed on an "AS IS" BASIS,
+#   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+#   See the License for the specific language governing permissions and
+#   limitations under the License.
+#
+
+from hopsworks_common.client.online_store_rest_client import (
+    OnlineStoreRestClientSingleton,
+    get_instance,
+    init_or_reset_online_store_rest_client,
+)
+
+
+__all__ = [
+    OnlineStoreRestClientSingleton,
+    get_instance,
+    init_or_reset_online_store_rest_client,
+]
diff --git a/python/hsfs/client/__init__.py b/python/hsfs/client/__init__.py
index 4ff5e8248..19e0feb8d 100644
--- a/python/hsfs/client/__init__.py
+++ b/python/hsfs/client/__init__.py
@@ -15,14 +15,26 @@
 #
 
 from hopsworks_common.client import (
+    auth,
+    base,
+    exceptions,
+    external,
     get_instance,
+    hopsworks,
     init,
+    online_store_rest_client,
     stop,
 )
 
 
 __all__ = [
+    auth,
+    base,
+    exceptions,
+    external,
     get_instance,
+    hopsworks,
     init,
+    online_store_rest_client,
     stop,
 ]