correct WT loading with b, dim

Author: Richard Michael

src/poli/objective_repository/dms_gb1/isolated_function.py

@@ -88,7 +88,7 @@ def __init__(
 
         self.dms_df = self._load_dms_data()
 
-        self.x0 = np.array(list(self.dms_df.iloc[self.wt].value))
+        self.x0 = np.array(list(self.wt))[None,:]
 
     def _load_dms_data(self) -> pd.DataFrame:
         return pd.read_csv(THIS_DIR / "assets" / "fitness.csv")

src/poli/objective_repository/dms_trpb/isolated_function.py

@@ -77,7 +77,7 @@ def __init__(
 
         self.dms_df = self._load_dms_data()
 
-        self.x0 = np.array(list(self.dms_df.iloc[self.wt].value))
+        self.x0 = np.array(list(self.wt))[None,:]
 
     def _load_dms_data(self) -> pd.DataFrame:
         return pd.read_csv(THIS_DIR / "assets" / "fitness.csv")