lalithadevi commited on
Commit
4e10a9e
1 Parent(s): 404c27a

Update db_operations/db_write.py

Browse files
Files changed (1) hide show
  1. db_operations/db_write.py +1 -1
db_operations/db_write.py CHANGED
@@ -1,6 +1,6 @@
1
  import pymongo
2
  import os
3
- from config import DATABASE, COLLECTION
4
  from logger import get_logger
5
 
6
  logger = get_logger()
 
1
  import pymongo
2
  import os
3
+ from config import DATABASE, COLLECTION, PREDICTION_DATABASE, PREDICTION_COLLECTION
4
  from logger import get_logger
5
 
6
  logger = get_logger()