Fix bug in extract_work_distributions

I was referring to `dataset` by `ds` accidentally.
choderalab · Mar 23, 2018 · 16125fb · 16125fb
1 parent 114514d
commit 16125fb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/protons/app/analysis.py b/protons/app/analysis.py
@@ -119,7 +119,7 @@ def extract_work_distributions(dataset: netCDF4.Dataset, state1_idx: int, state2
     as the forward distribution.    
     """
 
-    ncmc = ds["Protons/NCMC"]
+    ncmc = dataset["Protons/NCMC"]
     forward_work = []
     neg_reverse_work = []