remove __future__ annotations import for 3.6 compatibility

mbeissinger · mbeissinger · commit 19657237c08f · 2021-01-27T17:25:34.000-08:00
diff --git a/src/lobe/model/image_model.py b/src/lobe/model/image_model.py
@@ -1,7 +1,6 @@
 """
 Load a Lobe saved model for image classification
 """
-from __future__ import annotations
 from PIL import Image
 
 from .model import Model
@@ -14,7 +13,7 @@ class ImageModel(Model):
     signature: ImageClassificationSignature
 
     @classmethod
-    def load_from_signature(cls, signature: ImageClassificationSignature) -> ImageModel:
+    def load_from_signature(cls, signature: ImageClassificationSignature):
         # Select the appropriate backend
         model_format = signature.format
         if model_format == "tf":
@@ -30,7 +29,7 @@ def load_from_signature(cls, signature: ImageClassificationSignature) -> ImageMo
             raise ValueError(f"Model is an unsupported format: {model_format}")
 
     @classmethod
-    def load(cls, model_path: str) -> ImageModel:
+    def load(cls, model_path: str):
         # Load the signature
         return cls.load_from_signature(ImageClassificationSignature(model_path))
 
diff --git a/src/lobe/model/model.py b/src/lobe/model/model.py
@@ -1,5 +1,4 @@
-from __future__ import annotations
-from abc import ABC, abstractmethod 
+from abc import ABC, abstractmethod
 from ..signature import Signature
 
 
diff --git a/src/lobe/results.py b/src/lobe/results.py
@@ -1,4 +1,3 @@
-from __future__ import annotations
 import json
 from typing import List, Dict
 
@@ -20,7 +19,7 @@ class ClassificationResult:
     """
 
     @classmethod
-    def from_json(cls, json_str: str, labels: List[str] = None) -> ClassificationResult:
+    def from_json(cls, json_str: str, labels: List[str] = None):
         """
         Parse the classification results from a json string
         """
diff --git a/src/lobe/signature.py b/src/lobe/signature.py
@@ -1,4 +1,3 @@
-from __future__ import annotations
 import os
 import json
 import pathlib

Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,3 @@`
`1`		`-from __future__ import annotations`
`2`	`1`	`import os`
`3`	`2`	`import json`
`4`	`3`	`import pathlib`