From 4bda0671f68c439afb216028cd28d210f5b6fb5d Mon Sep 17 00:00:00 2001
From: devzhk <hzzheng@caltech.edu>
Date: Mon, 16 May 2022 16:58:34 -0700
Subject: [PATCH] 0.4.4 fix bug

---
 setup.cfg             | 2 +-
 src/CGDs/acgd.py      | 2 +-
 src/CGDs/cgd_utils.py | 2 +-
 3 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/setup.cfg b/setup.cfg
index a06844d..1aff517 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -1,6 +1,6 @@
 [metadata]
 name = CGDs
-version = 0.4.3
+version = 0.4.4
 author = Hongkai Zheng
 author_email = devzhk@gmail.com
 description = Adaptive Competitive Gradient Descent optimizer
diff --git a/src/CGDs/acgd.py b/src/CGDs/acgd.py
index 45ebc0e..31962b7 100644
--- a/src/CGDs/acgd.py
+++ b/src/CGDs/acgd.py
@@ -52,7 +52,7 @@ def set_lr(self, lr_max, lr_min):
         print('Maximizing side learning rate: {:.4f}\n '
               'Minimizing side learning rate: {:.4f}'.format(lr_max, lr_min))
 
-    def step(self, loss, trigger=None):
+    def step(self, loss, trigger=0.0):
         '''
         Update model weights using ACGD
         :param loss: objective function
diff --git a/src/CGDs/cgd_utils.py b/src/CGDs/cgd_utils.py
index 955d6a1..9ff2fa8 100644
--- a/src/CGDs/cgd_utils.py
+++ b/src/CGDs/cgd_utils.py
@@ -82,7 +82,7 @@ def conjugate_gradient(grad_x, grad_y,
 
 def Hvp_vec(grad_vec, params, vec,
             retain_graph=False,
-            trigger=None,
+            trigger=0.0,
             reducer=None,
             rebuild=False):
     '''