diff --git a/src/core_codemods/timezone_aware_datetime.py b/src/core_codemods/timezone_aware_datetime.py index d3d84592..bca4103a 100644 --- a/src/core_codemods/timezone_aware_datetime.py +++ b/src/core_codemods/timezone_aware_datetime.py @@ -74,9 +74,8 @@ def leave_Call(self, original_node: cst.Call, updated_node: cst.Call): return updated_node def _determine_module_and_kwarg(self, original_node: cst.Call): - maybe_name = self.get_aliased_prefix_name(original_node, self._module_name) - if maybe_name: + if maybe_name := self.get_aliased_prefix_name(original_node, self._module_name): # it's a regular import OR alias import if maybe_name == self._module_name: module = "datetime.datetime"