0.8.6

IndicoDataSolutions · Jan 10, 2020 · a7ffe78 · a7ffe78
1 parent f6bca1f
commit a7ffe78
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 6 deletions.
diff --git a/docs/conf.py b/docs/conf.py
@@ -28,10 +28,10 @@
 author = 'Madison May, Ben Townsend, Lily Zhang, Matthew Bayer'
 
 # The short X.Y version
-version, __version__ = ('0.8.5',) * 2
+version, __version__ = ('0.8.6',) * 2
 
 # The full version, including alpha/beta/rc tags
-release = '0.8.5'
+release = '0.8.6'
 
 # -- General configuration ---------------------------------------------------
 

diff --git a/finetune/__init__.py b/finetune/__init__.py
@@ -23,7 +23,7 @@
 from finetune.target_models.deployment_model import DeploymentModel
 from finetune.target_models.s2s import LMPred
 
-__version__, VERSION, version = ("0.8.5",) * 3
+__version__, VERSION, version = ("0.8.6",) * 3
 
 
 # Logging configuration

diff --git a/finetune/target_models/mtl.py b/finetune/target_models/mtl.py
@@ -52,7 +52,7 @@ def dataset_size(self):
         return self.dataset_size_
 
     def get_train_input_fns(
-        self, Xs, Y=None, batch_size=None, val_size=None, context=None
+        self, Xs, Y=None, batch_size=None, val_size=None, context=None, update_hook=None
     ):
         val_funcs = {}
         val_sizes = {}
@@ -64,7 +64,7 @@ def get_train_input_fns(
         for task_name in self.config.tasks:
             input_pipelines[task_name] = self.config.tasks[task_name]._get_input_pipeline(self)
             task_tuple = input_pipelines[task_name].get_train_input_fns(
-                Xs[task_name], Y[task_name], batch_size=batch_size, val_size=val_size, context=context
+                Xs[task_name], Y[task_name], batch_size=batch_size, val_size=val_size, context=context, update_hook=None
             )
             self.dataset_size_ += self.config.dataset_size
             frequencies.append(self.config.dataset_size)

diff --git a/setup.py b/setup.py
@@ -27,7 +27,7 @@
 setup(
     name="finetune",
     packages=find_packages(),
-    version="0.8.5",
+    version="0.8.6",
     install_requires=REQUIREMENTS,
     extras_require={
         "tf": ["tensorflow==1.14.0"],