From 12153256c3255d14e194beec4b3e621901401629 Mon Sep 17 00:00:00 2001
From: Fabio Buso <fabio@logicalclocks.com>
Date: Wed, 2 Dec 2020 14:15:39 +0100
Subject: [PATCH] [HOPSWORKS-2183] Add feature group, training dataset, storage
 connector counters to FeaturestoreDTO summary (#176)

---
 python/hsfs/feature_store.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/python/hsfs/feature_store.py b/python/hsfs/feature_store.py
index 9e6913650d..417161cec1 100644
--- a/python/hsfs/feature_store.py
+++ b/python/hsfs/feature_store.py
@@ -51,6 +51,9 @@ def __init__(
         offline_featurestore_name,
         hive_endpoint,
         online_enabled,
+        num_feature_groups=None,
+        num_training_datasets=None,
+        num_storage_connectors=None,
         online_featurestore_name=None,
         mysql_server_endpoint=None,
         online_featurestore_size=None,
@@ -69,6 +72,9 @@ def __init__(
         self._hive_endpoint = hive_endpoint
         self._mysql_server_endpoint = mysql_server_endpoint
         self._online_enabled = online_enabled
+        self._num_feature_groups = num_feature_groups
+        self._num_training_datasets = num_training_datasets
+        self._num_storage_connectors = num_storage_connectors
 
         self._feature_group_api = feature_group_api.FeatureGroupApi(self._id)
         self._storage_connector_api = storage_connector_api.StorageConnectorApi(