diff --git a/assets/src/ba_data/python/_ba.py b/assets/src/ba_data/python/_ba.py index dbdae18c..8ad46bfd 100644 --- a/assets/src/ba_data/python/_ba.py +++ b/assets/src/ba_data/python/_ba.py @@ -34,7 +34,7 @@ NOTE: This file was autogenerated by gendummymodule; do not edit by hand. """ # (hash we can use to see if this file is out of date) -# SOURCES_HASH=19018950468850503638731149460882909299 +# SOURCES_HASH=13501269283550798144571003636886602232 # I'm sorry Pylint. I know this file saddens you. Be strong. # pylint: disable=useless-suppression @@ -1998,7 +1998,7 @@ def get_display_resolution() -> Tuple[int, int]: return (0, 0) -def get_foreground_host_activity() -> ba.Activity: +def get_foreground_host_activity() -> Optional[ba.Activity]: """get_foreground_host_activity() -> Optional[ba.Activity] (internal) @@ -2010,7 +2010,7 @@ def get_foreground_host_activity() -> ba.Activity: return ba.Activity({}) -def get_foreground_host_session() -> ba.Session: +def get_foreground_host_session() -> Optional[ba.Session]: """get_foreground_host_session() -> Optional[ba.Session] (internal) diff --git a/assets/src/ba_data/python/ba/_apputils.py b/assets/src/ba_data/python/ba/_apputils.py index 2b974b87..c7a7bd73 100644 --- a/assets/src/ba_data/python/ba/_apputils.py +++ b/assets/src/ba_data/python/ba/_apputils.py @@ -321,8 +321,9 @@ def call_after_ad(call: Callable[[], Any]) -> None: if have_pro(): show = False # Pro disables interstitials. try: - is_tournament = (_ba.get_foreground_host_session().tournament_id is - not None) + session = _ba.get_foreground_host_session() + assert session is not None + is_tournament = session.tournament_id is not None except Exception: is_tournament = False if is_tournament: diff --git a/assets/src/ba_data/python/ba/_benchmark.py b/assets/src/ba_data/python/ba/_benchmark.py index 0a0c915e..c5a5aa08 100644 --- a/assets/src/ba_data/python/ba/_benchmark.py +++ b/assets/src/ba_data/python/ba/_benchmark.py @@ -147,7 +147,9 @@ def _reset_stress_test(args: Dict[str, Any]) -> None: from ba._enums import TimeType _ba.set_stress_testing(False, args['player_count']) _ba.screenmessage('Resetting stress test...') - _ba.get_foreground_host_session().end() + session = _ba.get_foreground_host_session() + assert session is not None + session.end() _ba.timer(1.0, Call(start_stress_test, args), timetype=TimeType.REAL)