bump version [ci skip]

kwatcharasupat · Jan 2, 2022 · 8e4306f · 8e4306f
1 parent e2fb8f4
commit 8e4306f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/setup.cfg b/setup.cfg
@@ -1,6 +1,6 @@
 [metadata]
 name = latte-metrics
-version = 0.0.1-alpha4
+version = 0.0.1-alpha5
 description = Latte: Cross-framework Python Package for Evaluation of Latent-based Generative Models
 long_description = file: README.md
 long_description_content_type = text/markdown

diff --git a/src/latte/metrics/keras/wrapper.py b/src/latte/metrics/keras/wrapper.py
@@ -13,7 +13,6 @@
 
 from ...metrics.base import LatteMetric
 
-
 def tf_to_numpy(args, kwargs):
     args = [a.numpy() for a in args]
     kwargs = {k: kwargs[k].numpy() for k in kwargs}
@@ -50,10 +49,11 @@ def update_state(self, *args, **kwargs):
         args, kwargs = tf_to_numpy(args, kwargs)
         self.metric.update_state(*args, **kwargs)
 
+    @tf.autograph.experimental.do_not_convert
     def result(self):
         return numpy_to_tf(self.metric.compute())
 
-    def reset_state(self):
+    def reset_stated(self):
         return self.metric.reset_state()
 
     def __getattr__(self, name: str):