FIX: Raise FileNotFoundError if in interactive mode

d-krupke · Feb 27, 2024 · 2722fc2 · 2722fc2
1 parent f742aad
commit 2722fc2
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/slurminade/function_map.py b/src/slurminade/function_map.py
@@ -133,6 +133,9 @@ def set_entry_point(entry_point: typing.Union[str, pathlib.Path]) -> None:
 def get_entry_point() -> Path:
  if FunctionMap.entry_point is None:
  import __main__
+ # check if attribute __file__ is available
+ if not hasattr(__main__, "__file__"):
+ raise FileNotFoundError("No entry point known.")
 
  entry_point = __main__.__file__