Adopt pymysql

py_experimenter/database_connector.py

@@ -333,6 +333,10 @@ def get_codecarbon_table(self) -> pd.DataFrame:
     def get_table(self, table_name: Optional[str] = None) -> pd.DataFrame:
         connection = self.connect()
         query = f"SELECT * FROM {self.table_name}" if table_name is None else f"SELECT * FROM {table_name}"
-        df = pd.read_sql(query, connection)
+        #suppress warning for pandas
+        import warnings
+        with warnings.catch_warnings():
+            warnings.filterwarnings("ignore", category=UserWarning)
+            df = pd.read_sql(query, connection)
         self.close_connection(connection)
         return df

py_experimenter/database_connector_mysql.py

@@ -3,7 +3,7 @@
 from typing import List, Tuple
 
 import numpy as np
-from mysql.connector import Error, connect
+from pymysql import Error, connect
 
 from py_experimenter.database_connector import DatabaseConnector
 from py_experimenter.exceptions import DatabaseConnectionError, DatabaseCreationError
@@ -62,7 +62,8 @@ def connect(self, credentials=None):
             raise DatabaseConnectionError(err)
 
     def _start_transaction(self, connection, readonly=False):
-        connection.start_transaction(readonly=readonly)
+        if not readonly:
+            connection.begin()
 
     def _table_exists(self, cursor, table_name:str = None) -> bool:
         table_name = table_name if table_name is not None else self.table_name

pyproject.toml

@@ -31,10 +31,10 @@ classifiers = [
 python = "^3.9"
 numpy = ">=1.15"
 pandas = ">=1.0"
-mysql-connector-python = ">=8.0"
 jupyterlab = "^3.5.0"
 joblib = "^1.2.0"
 codecarbon = "^2.2.1"
+pymysql = "^1.0.3"
 
 [tool.poetry.group.dev.dependencies]
 pytest = ">=7.0"

test/test_run_experiments/test_run_mysql_experiment.py

@@ -3,7 +3,7 @@
 from math import cos, sin
 
 import pandas as pd
-from mysql.connector.errors import ProgrammingError
+from pymysql.err import ProgrammingError
 
 from py_experimenter.experimenter import PyExperimenter
 from py_experimenter.result_processor import ResultProcessor

test/test_run_experiments/test_run_sqlite_experiment.py

@@ -4,7 +4,7 @@
 from math import cos, sin
 
 import pandas as pd
-from mysql.connector.errors import ProgrammingError
+from pymysql.err import ProgrammingError
 
 from py_experimenter.experimenter import PyExperimenter
 from py_experimenter.result_processor import ResultProcessor