Address the 'DataFrame' object has no attribute 'iteritems' error

judithabk6 · Mar 25, 2024 · 280d11e · 280d11e
1 parent 3eaea9a
commit 280d11e
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -17,7 +17,7 @@
     ),
     package_dir={"": "src"},
     install_requires=[
-        'pandas==1.2.1',
+        'pandas>=1.2.1',
         'scikit-learn>=0.22.1',
         'numpy>=1.19.2',
         'rpy2>=2.9.4',

diff --git a/src/med_bench/utils/utils.py b/src/med_bench/utils/utils.py
@@ -1,4 +1,5 @@
 import numpy as np
+import pandas as pd
 import rpy2.robjects as robjects
 
 import subprocess
@@ -55,6 +56,10 @@ def wrapper(*args, **kwargs):
                 )
                 return None
 
+            # To get rid of the 'DataFrame' object has no attribute 'iteritems' error due to pandas version mismatch in rpy2
+            # https://stackoverflow.com/a/76404841
+            pd.DataFrame.iteritems = pd.DataFrame.items
+
             for package in required_packages:
                 if not check_r_package(package):
                     if package != 'plmed':