Add __repr__ for AstronomicalObject

kboone · Apr 23, 2019 · 9bc126a · 9bc126a
1 parent d93981e
commit 9bc126a
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/avocado/astronomical_object.py b/avocado/astronomical_object.py
@@ -102,3 +102,7 @@ def plot_light_curve(self, data_only=False):
         plt.xlabel('Time (days)')
         plt.ylabel('Flux')
         plt.tight_layout()
+
+    def __repr__(self):
+        return "AstronomicalObject(object_id=%s)" % self.metadata['object_id']
+
diff --git a/avocado/dataset.py b/avocado/dataset.py
@@ -44,6 +44,7 @@ def __init__(self, name, metadata, observations=None):
                 observations.groupby('object_id'):
             meta_index = self.metadata.index.get_loc(object_id)
             object_metadata = meta_dicts[meta_index]
+            object_metadata['object_id'] = object_id
             new_object = AstronomicalObject(object_metadata,
                                             object_observations)
             objects.append(new_object)