diff --git a/tests/unit/cli/vyper_compile/test_compile_files.py b/tests/unit/cli/vyper_compile/test_compile_files.py index 03b3a1aabd..2a16efa777 100644 --- a/tests/unit/cli/vyper_compile/test_compile_files.py +++ b/tests/unit/cli/vyper_compile/test_compile_files.py @@ -42,7 +42,7 @@ def foo() -> FooStruct: @external def bar(a: address) -> FooStruct: - return {}.at(a).bar() + return {}(a).bar() """ BAR_CODE = """ diff --git a/vyper/semantics/types/module.py b/vyper/semantics/types/module.py index f2f9eb6422..f6ecbeb0bb 100644 --- a/vyper/semantics/types/module.py +++ b/vyper/semantics/types/module.py @@ -19,10 +19,7 @@ def __init__(self, module: vy_ast.Module, name: Optional[str] = None): # for function collisions interface_t = self.interface - # module.at(
) - at = MemberFunctionT(self, "at", [AddressT()], interface_t, is_modifying=False) - - members = {"at": at} + members = {} for f in self.functions: members[f.name] = f._metadata["type"] @@ -32,6 +29,12 @@ def __init__(self, module: vy_ast.Module, name: Optional[str] = None): def get_type_member(self, key: str, node: vy_ast.VyperNode) -> "VyperType": return self._helper.get_member(key, node) + def _ctor_call_return(self, node): + return self.interface._ctor_call_return(node) + + def _ctor_arg_types(self, node): + return self.interface._ctor_arg_types(node) + # this is a property, because the function set changes after AST expansion @property def functions(self):