DPDCE: fix Adapt init and pickle usage

2 files changed, 7 insertions, 7 deletions

diff --git a/dpd/apply_adapt_dumps.py b/dpd/apply_adapt_dumps.py
index f1f3359..0eb65f3 100755
--- a/dpd/apply_adapt_dumps.py
+++ b/dpd/apply_adapt_dumps.py
@@ -49,7 +49,7 @@ port_rc = cli_args.rc_port
 coef_path = cli_args.coefs
 filename = cli_args.file
 
-adapt = Adapt(port_rc, coef_path)
+adapt = Adapt.Adapt(port_rc, coef_path)
 
 print("Loading and applying DPD settings from {}".format(filename))
 adapt.load(filename)
diff --git a/dpd/src/Adapt.py b/dpd/src/Adapt.py
index c78c920..939e50e 100644
--- a/dpd/src/Adapt.py
+++ b/dpd/src/Adapt.py
@@ -239,20 +239,20 @@ class Adapt:
             "digital_gain": self.get_digital_gain(),
             "predistorter": self.get_predistorter()
         }
-        with open(path, "w") as f:
+        with open(path, "wb") as f:
             pickle.dump(d, f)
 
         return path
 
     def load(self, path):
         """Restore settings from a file"""
-        with open(path, "r") as f:
+        with open(path, "rb") as f:
             d = pickle.load(f)
 
-        self.set_txgain(d["txgain"])
-        self.set_digital_gain(d["digital_gain"])
-        self.set_rxgain(d["rxgain"])
-        self.set_predistorter(d["predistorter"])
+            self.set_txgain(d["txgain"])
+            self.set_digital_gain(d["digital_gain"])
+            self.set_rxgain(d["rxgain"])
+            self.set_predistorter(d["predistorter"])
 
 # The MIT License (MIT)
 #