Upload 2 files

__init__.py

@@ -0,0 +1,7 @@
+from modelscan.models.h5.scan import H5Scan
+from modelscan.models.pickle.scan import (
+    PickleScan,
+    NumpyScan,
+    PyTorchScan,
+)
+from modelscan.models.saved_model.scan import SavedModelScan

scan.py

@@ -0,0 +1,25 @@
+import abc
+from pathlib import Path
+from typing import List, Tuple, Union, Optional, IO
+
+from modelscan.error import Error
+from modelscan.issues import Issue
+
+
+class ScanBase(metaclass=abc.ABCMeta):
+    @staticmethod
+    @abc.abstractmethod
+    def name() -> str:
+        raise NotImplementedError
+
+    @staticmethod
+    @abc.abstractmethod
+    def scan(
+        source: Union[str, Path], data: Optional[IO[bytes]] = None
+    ) -> Tuple[List[Issue], List[Error]]:
+        raise NotImplementedError
+
+    @staticmethod
+    @abc.abstractmethod
+    def supported_extensions() -> List[str]:
+        raise NotImplementedError