mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-27 09:08:14 +00:00
Fix lint errors: remove unused imports
- Remove unused typing.Optional import - Remove unused unittest.mock.MagicMock import - Add noqa comment for necessary mlflow import in availability check Co-Authored-By: João <joao@crewai.com>
This commit is contained in:
@@ -1,7 +1,6 @@
|
|||||||
"""MLFlow integration utilities for CrewAI."""
|
"""MLFlow integration utilities for CrewAI."""
|
||||||
|
|
||||||
import logging
|
import logging
|
||||||
from typing import Optional
|
|
||||||
|
|
||||||
logger = logging.getLogger(__name__)
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
@@ -9,7 +8,7 @@ logger = logging.getLogger(__name__)
|
|||||||
def is_mlflow_available() -> bool:
|
def is_mlflow_available() -> bool:
|
||||||
"""Check if MLFlow is available."""
|
"""Check if MLFlow is available."""
|
||||||
try:
|
try:
|
||||||
import mlflow
|
import mlflow # noqa: F401
|
||||||
return True
|
return True
|
||||||
except ImportError:
|
except ImportError:
|
||||||
return False
|
return False
|
||||||
|
|||||||
@@ -1,7 +1,7 @@
|
|||||||
"""Tests for MLFlow integration."""
|
"""Tests for MLFlow integration."""
|
||||||
|
|
||||||
import pytest
|
import pytest
|
||||||
from unittest.mock import Mock, patch, MagicMock
|
from unittest.mock import Mock, patch
|
||||||
import sys
|
import sys
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user