Update checker using safe_infer

Author: Peter Hamfelt

pylint_ml/checkers/numpy/numpy_nan_comparison.py

@@ -31,7 +31,7 @@ class NumpyNaNComparisonChecker(LibraryBaseChecker):
     @classmethod
     def __is_np_nan_call(cls, node: nodes.Attribute) -> bool:
         """Check if the node represents a call to np.nan."""
-        return node.attrname in NUMPY_NAN and (infer_specific_module_from_attribute(node=node, module_name="numpy"))
+        return node.attrname in NUMPY_NAN and (infer_specific_module_from_attribute(node=node, module_name=NUMPY))
 
     @only_required_for_messages("numpy-nan-compare")
     def visit_compare(self, node: nodes.Compare) -> None:

pylint_ml/checkers/pandas/pandas_dataframe_iterrows.py

@@ -12,6 +12,7 @@
 
 from pylint_ml.checkers.config import PANDAS
 from pylint_ml.checkers.library_base_checker import LibraryBaseChecker
+from pylint_ml.checkers.utils import infer_specific_module_from_attribute
 
 
 class PandasIterrowsChecker(LibraryBaseChecker):
@@ -30,7 +31,10 @@ def visit_call(self, node: nodes.Call) -> None:
         if not self.is_library_imported_and_version_valid(lib_name=PANDAS, required_version=None):
             return
 
-        if isinstance(node.func, nodes.Attribute):
+        if (
+                isinstance(node.func, nodes.Attribute)
+                and infer_specific_module_from_attribute(node=node.func, module_name=PANDAS)
+        ):
             method_name = getattr(node.func, "attrname", None)
             if method_name == "iterrows":
                 object_name = getattr(node.func.expr, "name", None)