temp fixes

pyupio · Sep 16, 2024 · 958a39e · 958a39e
1 parent 333421d
commit 958a39e
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 5 deletions.
diff --git a/safety/scan/command.py b/safety/scan/command.py
@@ -45,8 +45,6 @@
 class ScannableEcosystems(Enum):
     """Enum representing scannable ecosystems."""
     PYTHON = Ecosystem.PYTHON.value
-    PYPROJECT_TOML = Ecosystem.PYPROJECT_TOML.value
-
 
 def process_report(
     obj: Any, console: Console, report: ReportModel, output: str,

diff --git a/safety/scan/finder/handlers.py b/safety/scan/finder/handlers.py
@@ -52,7 +52,7 @@ def can_handle(self, root: str, file_name: str, include_files: Dict[FileType, Li
         return None
 
     @abstractmethod
-    def download_required_assets(self, session) -> Dict[str, str]:
+    def download_required_assets(self, session):
         """
         Abstract method to download required assets for handling files. Should be implemented
         by subclasses.
@@ -116,7 +116,7 @@ def download_required_assets(self, session):
 class PyProjectTomlHandler(FileHandler):
     def __init__(self) -> None:
         super().__init__()
-        self.ecosystem = Ecosystem.PYPROJECT_TOML
+        self.ecosystem = Ecosystem.PYTHON
 
     def download_required_assets(self, session):
         from safety.safety import fetch_database
@@ -174,5 +174,5 @@ def download_required_assets(self, session) -> None:
 ECOSYSTEM_HANDLER_MAPPING = MappingProxyType({
     Ecosystem.PYTHON: PythonFileHandler,
     Ecosystem.SAFETY_PROJECT: SafetyProjectFileHandler,
-    Ecosystem.PYPROJECT_TOML: PyProjectTomlHandler,
+    # Ecosystem.PYPROJECT_TOML: PyProjectTomlHandler,
 })