diff --git a/submarine-sdk/pysubmarine/submarine/ml/pytorch/model/ctr/__init__.py b/submarine-sdk/pysubmarine/submarine/ml/pytorch/model/ctr/__init__.py index ac6fd3f574..7ebc1f4779 100644 --- a/submarine-sdk/pysubmarine/submarine/ml/pytorch/model/ctr/__init__.py +++ b/submarine-sdk/pysubmarine/submarine/ml/pytorch/model/ctr/__init__.py @@ -14,6 +14,6 @@ # limitations under the License. from .deepfm import DeepFM -from .afm import AttentionalFM +from .afm import AttentionalFM __all__ = ['DeepFM', 'AttentionalFM']