fix tests

src/python/nimbusml/tests/timeseries/test_rollingwindow.py

@@ -32,10 +32,10 @@ def test_simple_rolling_window(self):
                            horizon=2)
         result = rw.fit_transform(df)
 
-        self.assertTrue(math.isnan(result.loc[0, 'ts_r']))
-        self.assertEqual(result.loc[1, 'ts_r'], 1)
-        self.assertEqual(result.loc[2, 'ts_r'], 3)
-        self.assertEqual(result.loc[3, 'ts_r'], 5)
+        self.assertTrue(math.isnan(result.loc[0, 'ts_r.1']))
+        self.assertEqual(result.loc[1, 'ts_r.1'], 1)
+        self.assertEqual(result.loc[2, 'ts_r.1'], 3)
+        self.assertEqual(result.loc[3, 'ts_r.1'], 5)
 
     def test_simple_rolling_window2(self):
 

src/python/tests_extended/test_export_to_onnx.py

@@ -118,10 +118,12 @@
     'OnnxRunner',
     'Sentiment',
     'TensorFlowScorer',
+    'TimeSeriesImputer',
     'TreeFeaturizer',
     'WordEmbedding',
     'Binner',
     'BootstrapSampler',
+    'DateTimeSplitter',
     'EnsembleClassifier',
     'EnsembleRegressor',
     'FactorizationMachineBinaryClassifier',
@@ -138,12 +140,18 @@
     'RangeFilter',
     'Resizer',
     'RobustScaler',
+    'RollingWindow',
+    'ForecastingPivot',
+    'LagLeadOperator',
+    'ShortDrop',
     'SkipFilter',
     'SsaChangePointDetector',
     'SsaForecaster',
     'SsaSpikeDetector',
     'SymSgdBinaryClassifier',
     'TakeFilter',
+    'ToKeyImputer',
+    'ToString',
     'EnsembleClassifier',
     'EnsembleRegressor',
     'CharTokenizer',