Skip to content

Commit ff849d9

Browse files
rstzcopybara-github
authored andcommitted
Implement shell of get_config to silence warnings
PiperOrigin-RevId: 579758350
1 parent 244576e commit ff849d9

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

tensorflow_decision_forests/keras/core_inference.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -413,6 +413,10 @@ def make_inspector(self, index: int = 0) -> inspector_lib.AbstractInspector:
413413
path, file_prefix=self.yggdrasil_model_prefix(index)
414414
)
415415

416+
def get_config(self):
417+
"""Not supported by TF-DF, returning empty directory to avoid warnings."""
418+
return {}
419+
416420
@tf.function(input_signature=[])
417421
def yggdrasil_model_path_tensor(
418422
self, multitask_model_index: int = 0

0 commit comments

Comments
 (0)