Skip to content

Commit

Permalink
Comment out DummyModule
Browse files Browse the repository at this point in the history
  • Loading branch information
kumaranvpl committed Jan 17, 2025
1 parent c2bc4c5 commit a4b5e1f
Showing 1 changed file with 11 additions and 12 deletions.
23 changes: 11 additions & 12 deletions autogen/handle_imports.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,10 +3,9 @@
# SPDX-License-Identifier: Apache-2.0

import importlib
import sys
from contextlib import contextmanager
from functools import wraps
from typing import Any, Generator, Iterable, Union
from typing import Generator, Iterable, Union

from autogen.exception_utils import AutogenImportError

Expand All @@ -23,15 +22,15 @@ def check_for_missing_imports() -> Generator[None, None, None]:
pass # Ignore ImportErrors during this context


class DummyModule:
"""A dummy module that raises ImportError when any attribute is accessed"""
# class DummyModule:
# """A dummy module that raises ImportError when any attribute is accessed"""

def __init__(self, name: str, dep_target: str):
self._name = name
self._dep_target = dep_target
# def __init__(self, name: str, dep_target: str):
# self._name = name
# self._dep_target = dep_target

def __getattr__(self, attr: str) -> Any:
raise AutogenImportError(missing_modules=self._name, dep_target=self._dep_target)
# def __getattr__(self, attr: str) -> Any:
# raise AutogenImportError(missing_modules=self._name, dep_target=self._dep_target)


def requires_optional_import(modules: Union[str, Iterable[str]], dep_target: str):
Expand All @@ -55,9 +54,9 @@ def decorator(cls):
except ImportError:
missing_modules.append(module_name)
# Create dummy module
dummy_module = DummyModule(module_name, dep_target)
dummy_modules[module_name] = dummy_module
sys.modules[module_name] = dummy_module
# dummy_module = DummyModule(module_name, dep_target)
# dummy_modules[module_name] = dummy_module
# sys.modules[module_name] = dummy_module

if missing_modules:
# Replace real class with dummy that raises ImportError
Expand Down

0 comments on commit a4b5e1f

Please sign in to comment.