add asserts for hidden attributes

modal-labs · Feb 13, 2024 · 255cad3 · 255cad3
1 parent 16ed523
commit 255cad3
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/modal/functions.py b/modal/functions.py
@@ -1016,6 +1016,7 @@ async def lookup(
     @property
     def tag(self):
         """mdmd:hidden"""
+        assert hasattr(self, "_tag")
         return self._tag
 
     @property
@@ -1032,6 +1033,7 @@ def env(self) -> FunctionEnv:
 
     def get_build_def(self) -> str:
         """mdmd:hidden"""
+        assert hasattr(self, "_raw_f") and hasattr(self, "_build_args")
         return f"{inspect.getsource(self._raw_f)}\n{repr(self._build_args)}"
 
     # Live handle methods