Fixed yaml.load warnings

grmpy/read/read.py

@@ -1,9 +1,8 @@
 """The module contains the main function of the init file import process."""
 import yaml
 
-from grmpy.read.read_auxiliary import create_attr_dict_sim
-from grmpy.read.read_auxiliary import create_attr_dict_est
 from grmpy.check.check import check_presence_init
+from grmpy.read.read_auxiliary import create_attr_dict_est, create_attr_dict_sim
 
 
 def read(file, semipar=False, include_constant=False):
@@ -15,7 +14,7 @@ def read(file, semipar=False, include_constant=False):
 
     # Load the initialization file
     with open(file) as y:
-        init_dict = yaml.load(y)
+        init_dict = yaml.load(y, Loader=yaml.FullLoader)
 
     # Process the initialization file
     attr_dict = create_attr_dict_est(init_dict, semipar, include_constant)
@@ -32,7 +31,7 @@ def read_simulation(file):
 
     # Load the initialization file
     with open(file) as y:
-        init_dict = yaml.load(y)
+        init_dict = yaml.load(y, Loader=yaml.FullLoader)
 
     # Process the initialization file
     attr_dict = create_attr_dict_sim(init_dict)

grmpy/test/test_unit.py

@@ -1,6 +1,7 @@
 """The module provides unit tests for different aspects of the simulation process."""
 import numpy as np
 import pandas as pd
+import pytest
 
 from grmpy.check.auxiliary import read_data
 from grmpy.estimate.estimate import fit
@@ -354,6 +355,7 @@ def test12():
         np.testing.assert_equal(columns, real_column_values)
 
 
+@pytest.mark.filterwarnings("ignore::RuntimeWarning")
 def test13():
     """This test checks if functions that affect the estimation output adjustment work as
     intended.