Update doctests

Author: yngvem

tlviz/_xarray_wrapper.py

@@ -57,12 +57,12 @@ def add_factor_metadata(cp_tensor, dataset):
     >>> bikes.coords
     Coordinates:
       * End station name  (End station name) object 2kB '7 Juni Plassen' ... 'Øke...
+        lat               (End station name) float64 2kB 59.92 59.93 ... 59.93 59.92
+        lon               (End station name) float64 2kB 10.73 10.75 ... 10.8 10.78
       * Year              (Year) int32 8B 2020 2021
       * Month             (Month) int32 48B 1 2 3 4 5 6 7 8 9 10 11 12
       * Day of week       (Day of week) int32 28B 0 1 2 3 4 5 6
       * Hour              (Hour) int32 96B 0 1 2 3 4 5 6 7 ... 17 18 19 20 21 22 23
-        lat               (End station name) float64 2kB 59.92 59.93 ... 59.93 59.92
-        lon               (End station name) float64 2kB 10.73 10.75 ... 10.8 10.78
 
     We see that the ``End station name`` dimension has two additional columns: ``lat`` and ``lon``.
     These contain metadata about the end station coordinates, and it can be useful to have these

tlviz/postprocessing.py

@@ -317,7 +317,7 @@ def postprocess(
         >>> print(type(cp_tensor[1][0]))
         <class 'numpy.ndarray'>
         >>> print(type(cp_tensor_postprocessed[1][0]))
-        <class 'pandas.core.frame.DataFrame'>
+        <class 'pandas.DataFrame'>
 
 
         We see that after postprocessing, the factor matrix has unit norm