import registry in __init__

building_dao/__init__.py

-from .building_generic import BuildingGeneric, VersionConflictError
\ No newline at end of file
+from .building_generic import BuildingGeneric, VersionConflictError
+from MaxIcsRegistry import MaxIcsRegistry
+
+registry = MaxIcsRegistry.from_env(staging=False)
\ No newline at end of file

building_dao/building_generic.py

@@ -6,21 +6,17 @@ from typing import Optional
 import requests
 from requests.exceptions import HTTPError
 
-from MaxIcsRegistry import MaxIcsRegistry
 from retrying import retry
 from simplejson import JSONDecodeError
+from . import registry
 
-# os.environ['MESOS_TASK_ID'] = "maxics_wind-vulnerability-0_google-street-view-image-poller.6f22b79d-73ae-11e8-8774-0242b35465ae"
-registry = MaxIcsRegistry.from_env(staging=False)
 building_center_info = registry.get_info("building-knowledge-information-center")
 
 MAX_RETRIES = 5  # max retries number
 WAIT_BETWEEN_RETRIES_IN_MS = 200  # time between each retry
 
-
 class VersionConflictError(Exception):
     """Raised when there is a version conflict when updating ES doc"""
-
     pass
 
 class BuildingGeneric: