Fix bug in UnobservableModel.plot_incumbent_best_answers.

manuelbieri · Nov 4, 2021 · e0b22e2 · e0b22e2
2 parents 150681f + 232e1cc
commit e0b22e2
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/Shelegia_Motta_2021/Models.py b/Shelegia_Motta_2021/Models.py
@@ -808,7 +808,7 @@ def __init__(self, u: float = 1, B: float = 0.5, small_delta: float = 0.5, delta
         super(UnobservableModel, self).__init__(u=u, B=B, small_delta=small_delta, delta=delta, K=K, beta=beta)
 
     def plot_incumbent_best_answers(self, axis: matplotlib.axes.Axes = None, **kwargs) -> matplotlib.axes.Axes:
-        return self.plot_equilibrium(axis=axis, kwargs=kwargs)
+        return self.plot_equilibrium(axis=axis, **kwargs)
 
     def _create_choice_answer_label(self, entrant: Literal["complement", "substitute", "indifferent"],
                                     incumbent: Literal["copy", "refrain"],
@@ -885,6 +885,6 @@ def get_copying_fixed_costs_values(self) -> Dict[str, float]:
 
 
 if __name__ == '__main__':
-    bargaining_power_model = Shelegia_Motta_2021.BargainingPowerModel(beta=0.6)
-    bargaining_power_model.plot_equilibrium()
+    bargaining_power_model = Shelegia_Motta_2021.UnobservableModel(beta=0.6)
+    bargaining_power_model.plot_incumbent_best_answers(options_legend=True)
     plt.show()