Skip to content

Commit

Permalink
change(ml): Deprecate AutoML model support (#670)
Browse files Browse the repository at this point in the history
* chore(ml): Deprecate AutoML model support

* fix lint
  • Loading branch information
lahirumaramba authored Feb 2, 2023
1 parent 18714fb commit 68001d9
Showing 1 changed file with 10 additions and 1 deletion.
11 changes: 10 additions & 1 deletion firebase_admin/ml.py
Original file line number Diff line number Diff line change
Expand Up @@ -24,6 +24,7 @@
import time
import os
from urllib import parse
import warnings

import requests

Expand Down Expand Up @@ -383,11 +384,14 @@ def __ne__(self, other):

@staticmethod
def _init_model_source(data):
"""Initialize the ML model source."""
gcs_tflite_uri = data.pop('gcsTfliteUri', None)
if gcs_tflite_uri:
return TFLiteGCSModelSource(gcs_tflite_uri=gcs_tflite_uri)
auto_ml_model = data.pop('automlModel', None)
if auto_ml_model:
warnings.warn('AutoML model support is deprecated and will be removed in the next '
'major version.', DeprecationWarning)
return TFLiteAutoMlSource(auto_ml_model=auto_ml_model)
return None

Expand Down Expand Up @@ -604,9 +608,14 @@ def as_dict(self, for_upload=False):


class TFLiteAutoMlSource(TFLiteModelSource):
"""TFLite model source representing a tflite model created with AutoML."""
"""TFLite model source representing a tflite model created with AutoML.
AutoML model support is deprecated and will be removed in the next major version.
"""

def __init__(self, auto_ml_model, app=None):
warnings.warn('AutoML model support is deprecated and will be removed in the next '
'major version.', DeprecationWarning)
self._app = app
self.auto_ml_model = auto_ml_model

Expand Down

0 comments on commit 68001d9

Please sign in to comment.