From fcbd50e414c58a9b7dc6265b18577095721252be Mon Sep 17 00:00:00 2001
From: Jongwook Choi <wookayin@gmail.com>
Date: Mon, 22 Oct 2018 00:21:22 -0400
Subject: [PATCH] Update requirements.txt: delegate to setup.py

---
 requirements.txt  | 8 ++++----
 tfplot/wrapper.py | 6 ++++--
 2 files changed, 8 insertions(+), 6 deletions(-)

diff --git a/requirements.txt b/requirements.txt
index 13dc7b3..c60efdc 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,6 +1,6 @@
-six>=1.10.0
-numpy>=1.10.4
-matplotlib>=2.0.0
-tensorflow>=1.0.0
+-e .
+
+# test/dev environment
+tensorflow>=1.4.0
 
 # vim: set ft=conf:
diff --git a/tfplot/wrapper.py b/tfplot/wrapper.py
index da114fc..30b0b65 100644
--- a/tfplot/wrapper.py
+++ b/tfplot/wrapper.py
@@ -195,7 +195,7 @@ def _wrapped_factory_fn(*args, **kwargs_call):
 
 @biwrap
 def autowrap(plot_func=REQUIRED, _sentinel=None,
-             name=None, figsize=None, tight_layout=False):
+             batch=False, name=None, figsize=None, tight_layout=False):
     """
     Wrap a function as a TensorFlow operation similar to :func:`tfplot.wrap()`
     (as a decorator or with normal function call), but provides with additional
@@ -228,6 +228,8 @@ def autowrap(plot_func=REQUIRED, _sentinel=None,
         has a parameter named ``fig`` and/or ``ax``, new instances of
         ``Figure`` and/or ``AxesSubplot`` will be created and passed.
 
+      batch: If True, all the tensors passed as argument will be
+        assumed to be batched. Default value is False.
       name: A default name for the operation (optional). If not given, the
         name of ``plot_func`` will be used.
 
@@ -285,7 +287,7 @@ def _wrapped_plot_fn(*args, **kwargs_call):
 
         return ret
 
-    _wrapped_fn = wrap(_wrapped_plot_fn, name=name)  # TODO kwargs
+    _wrapped_fn = wrap(_wrapped_plot_fn, batch=batch, name=name)  # TODO kwargs
 
     _wrapped_fn.__name__ = 'autowrap[%s]' % plot_func.__name__
     if hasattr(plot_func, '__qualname__'):