v0.6.5 revision

SSusantAchary · Nov 12, 2020 · 59dce12 · 59dce12
1 parent 434cc0d
commit 59dce12
Showing 1 changed file with 22 additions and 0 deletions.
diff --git a/fiftyone/migrations/revisions/v0_6_5.py b/fiftyone/migrations/revisions/v0_6_5.py
@@ -0,0 +1,22 @@
+"""
+FiftyOne v0.6.5 revision
+
+| Copyright 2017-2020, Voxel51, Inc.
+| `voxel51.com <https://voxel51.com/>`_
+|
+"""
+import pymongo as pm
+
+
+def up(db, dataset_name):
+    match_d = {"name": dataset_name}
+    dataset_dict = db.datasets.find_one(match_d)
+    sample_coll = db[dataset_dict["sample_collection_name"]]
+    sample_coll.update_many({}, {"$rename": {"media_type": "_media_type"}})
+
+
+def down(db, dataset_name):
+    match_d = {"name": dataset_name}
+    dataset_dict = db.datasets.find_one(match_d)
+    sample_coll = db[dataset_dict["sample_collection_name"]]
+    sample_coll.update_many({}, {"$rename": {"_media_type": "media_type"}})