HEL/midas
1
0
Fork 0
Code Issues Pull Requests Actions Releases 2 Activity

fix(checker): allow calling unknown method on dataframes

Browse Source
This commit is contained in:
Louis Heredero
2026-06-29 11:01:53 +02:00
parent db40198357
commit 75bd203d4a
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
midas/checker/frame_methods.py
View File

@@ -42,7 +42,7 @@ class Call:
class _MethodRegistryMeta(type): class _MethodRegistryMeta(type):
_methods: dict[str, Callable] = {} _methods: dict[str, Callable[..., Type]] = {}
def __new__( def __new__(
cls, cls,
@@ -55,7 +55,7 @@ class _MethodRegistryMeta(type):
for attr in namespace.values(): for attr in namespace.values():
if callable(attr) and hasattr(attr, "__method_names__"): if callable(attr) and hasattr(attr, "__method_names__"):
for name in attr.__method_names__: # type: ignore for name in attr.__method_names__: # type: ignore
new_class._methods[name] = attr new_class._methods[name] = attr # type: ignore
return new_class return new_class
@@ -76,9 +76,9 @@ class MethodRegistry(metaclass=_MethodRegistryMeta):
method: str, method: str,
call: Call, call: Call,
) -> Type: ) -> Type:
func: Optional[Callable] = self._methods.get(method) func: Optional[Callable[..., Type]] = self._methods.get(method)
if func is None: if func is None:
self.reporter.error(call.location, f"Unknown method {method}") self.reporter.warning(call.location, f"Unknown method {method}")
return UnknownType() return UnknownType()
return func(self, call) return func(self, call)