Fix from_cesium_dataset docstring

rth · rth · commit 824ef494c0a8 · 2020-03-26T12:07:13.000+01:00
diff --git a/conftest.py b/conftest.py
@@ -26,7 +26,7 @@ def pytest_collection_modifyitems(config, items):
                 "tslearn.utils.to_cesium_dataset",
             ]:
                 item.add_marker(skip_marker)
-    elif cesium is None:
+    if cesium is None:
         skip_marker = pytest.mark.skip(reason="cesium not installed!")
         for item in items:
             if item.name in [
diff --git a/tslearn/utils.py b/tslearn/utils.py
@@ -1208,6 +1208,7 @@ def from_cesium_dataset(X):
 
     Examples
     --------
+    >>> from cesium.time_series import TimeSeries
     >>> cesium_ds = [TimeSeries(m=numpy.array([1, 2, 3, 4]))]
     >>> tslearn_arr = from_cesium_dataset(cesium_ds)
     >>> tslearn_arr.shape
