don't change behavior yet

src/pylint_airflow/checkers/xcom.py

@@ -12,7 +12,6 @@
 from typing import Set, Dict, Tuple
 
 import astroid
-from astroid import AttributeInferenceError
 from pylint import checkers
 from pylint.checkers import utils
 
@@ -78,12 +77,8 @@ def get_xcoms_from_tasks(
         if callable_func_name == "<lambda>":  # TODO support lambdas
             continue
 
-        try:
-            module_attribute = node.getattr(callable_func_name)
-        except AttributeInferenceError:
-            continue
-        else:
-            callable_func = module_attribute[0]
+        callable_func = node.getattr(callable_func_name)[0]
+        # ^ TODO: handle builtins and attribute imports that will raise on this call
 
         if not isinstance(callable_func, astroid.FunctionDef):
             continue  # Callable_func is str not FunctionDef when imported

tests/pylint_airflow/checkers/test_xcom.py

@@ -95,7 +95,6 @@ def test_should_detect_imported_callable_as_attribute(self):
 
         assert result == expected_result
 
-    @pytest.mark.xfail(reason="Not yet implemented", raises=AssertionError, strict=True)
     def test_should_detect_imported_callable_as_name(self):
         test_code = """
         from airflow.operators.python_operator import PythonOperator
@@ -162,6 +161,9 @@ def test_should_skip_lambda_callable(self):
 
         assert result == ({}, set())
 
+    @pytest.mark.xfail(
+        reason="Not yet implemented", raises=astroid.AttributeInferenceError, strict=True
+    )
     def test_should_skip_builtin_callable(self):
         test_code = """
         from airflow.operators.python_operator import PythonOperator
@@ -178,6 +180,9 @@ def test_should_skip_builtin_callable(self):
 
         assert result == ({}, set())
 
+    @pytest.mark.xfail(
+        reason="Not yet implemented", raises=astroid.AttributeInferenceError, strict=True
+    )
     def test_should_skip_imported_callable_as_attribute(self):
         test_code = """
         from airflow.operators.python_operator import PythonOperator