Update inputparser.py

Geodels · Mar 21, 2024 · e89a5e6 · e89a5e6
1 parent a2d0210
commit e89a5e6
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/gospl/tools/inputparser.py b/gospl/tools/inputparser.py
@@ -3,7 +3,7 @@
 import petsc4py
 import numpy as np
 import pandas as pd
-import ruamel.yaml as YAML
+from ruamel.yaml import YAML
 
 from operator import itemgetter
 from scipy.interpolate import interp1d
@@ -41,7 +41,8 @@ def __init__(self, filename):
 
         # Open YAML file
         with open(filename, "r") as finput:
-            self.input = YAML.load(finput, Loader=YAML.Loader)
+            yaml = YAML(typ='rt')
+            self.input = yaml.load(finput)
 
         if MPIrank == 0 and "name" in self.input.keys() and self.verbose:
             print(