Give assign layer's second parameter 'output' a default value: None. If
it is None, the output variable will be created inside the layer.