Minor changes to function arguments

privacy/bolton/models.py

@@ -60,7 +60,7 @@ class BoltonModel(Model):
     self._layers_instantiated = False
     self._dtype = dtype
 
-  def call(self, inputs, training=False):  # pylint: disable=arguments-differ
+  def call(self, inputs):  # pylint: disable=arguments-differ
     """Forward pass of network
 
     Args:

tutorials/bolton_tutorial.ipynb

@@ -321,7 +321,7 @@
     "                                                  kernel_regularizer=reg_layer\n",
     "                                                  )\n",
     "        \n",
-    "    def call(self, inputs, training=False):\n",
+    "    def call(self, inputs):\n",
     "        return self.output_layer(inputs)\n",
     "\n",
     "optimizer = tf.optimizers.SGD()\n",
@@ -420,10 +420,10 @@
   "pycharm": {
    "stem_cell": {
     "cell_type": "raw",
+    "source": [],
     "metadata": {
      "collapsed": false
-    },
-    "source": []
+    }
    }
   }
  },