Remove workaround as #112 is fixed now

skorch-dev · Nov 29, 2017 · 5184dd3 · 5184dd3
1 parent b5df830
commit 5184dd3
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/examples/word_language_model/train.py b/examples/word_language_model/train.py
@@ -50,11 +50,7 @@ def on_epoch_end(self, net, **kwargs):
 def my_train_split(X, y):
     # Return (corpus.train, corpus.valid) in case the network
     # is fitted using net.fit(corpus.train).
-    #
-    # TODO: remove dummy y values once #112 is fixed.
-    #
-    import numpy as np
-    return X, corpus.valid, np.zeros(len(X)), np.zeros(len(corpus.valid))
+    return X, corpus.valid, None, None
 
 net = Net(
     module=RNNModel,