Skip to content

Commit

Permalink
Reformat code
Browse files Browse the repository at this point in the history
  • Loading branch information
notoraptor committed Sep 19, 2022
1 parent 5cb4d35 commit 3b15585
Showing 1 changed file with 4 additions and 5 deletions.
9 changes: 4 additions & 5 deletions .github/workflows/orion/pickle_to_mongodb.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,5 @@
import sys
from orion.core.io.database.pickleddb import PickledDB
from orion.core.io.database.mongodb import MongoDB
from orion.core.io.database.pickleddb import PickledDB
from orion.storage.base import setup_storage


Expand All @@ -15,13 +14,13 @@ def main():
)
pickle_db = storage._db
assert isinstance(pickle_db, PickledDB)
mongo_db = MongoDB(name = 'orion_dashboard_test')
mongo_db = MongoDB(name="orion_dashboard_test")
with pickle_db.locked_database(write=False) as database:
for collection_name in database._db.keys():
print(f'[{collection_name}]')
print(f"[{collection_name}]")
data = database.read(collection_name)
mongo_db.write(collection_name, data)
print('Pickle to Mongo DB done.')
print("Pickle to Mongo DB done.")


if __name__ == "__main__":
Expand Down

0 comments on commit 3b15585

Please sign in to comment.