fix auto doc (#708)

logicalclocks · Jul 12, 2022 · 747dcd1 · 747dcd1
1 parent c2a86ca
commit 747dcd1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/auto_doc.py b/auto_doc.py
@@ -46,10 +46,10 @@
         "fg_create": ["hsfs.feature_store.FeatureStore.create_external_feature_group"],
         "fg_get": ["hsfs.feature_store.FeatureStore.get_external_feature_group"],
         "fg_properties": keras_autodoc.get_properties(
-            "hsfs.feature_group.OnDemandFeatureGroup"
+            "hsfs.feature_group.ExternalFeatureGroup"
         ),
         "fg_methods": keras_autodoc.get_methods(
-            "hsfs.feature_group.OnDemandFeatureGroup",
+            "hsfs.feature_group.ExternalFeatureGroup",
             exclude=[
                 "from_response_json",
                 "update_from_response_json",