rollback module.at() method

vyperlang · Nov 20, 2023 · 0a786f7 · 0a786f7
1 parent 4a8cc12
commit 0a786f7
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 5 deletions.
diff --git 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
@@ -19,10 +19,7 @@ def __init__(self, module: vy_ast.Module, name: Optional[str] = None):
         # for function collisions
         interface_t = self.interface
 
-        # module.at(<address>)
-        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):