Omit target attribute if all are None (instead of storing nans)

cesium-ml · Oct 3, 2016 · 163fa40 · 163fa40
1 parent 4783290
commit 163fa40
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/cesium/featurize.py b/cesium/featurize.py
@@ -135,7 +135,7 @@ def assemble_featureset(feature_dicts, time_series=None, targets=None,
     featureset = xr.Dataset(combined_feature_dict)
     if names is not None:
         featureset.coords['name'] = ('name', np.array(names))
-    if targets is not None:
+    if targets is not None and any(targets):
         featureset.coords['target'] = ('name', np.array(targets))
     return featureset
 

diff --git a/cesium/tests/test_featurize.py b/cesium/tests/test_featurize.py
@@ -288,4 +288,4 @@ def test_featurize_time_series_no_targets():
                                            scheduler=get_sync)
     npt.assert_array_equal(sorted(fset.data_vars),
                            ['amplitude', 'meta1', 'std_err'])
-    npt.assert_array_equal(fset.target.values, [None])
+    assert('target' not in fset)