22import os
33import sys
44import time
5+ from urllib .parse import urlparse
56
67import requests
7- from urllib .parse import urlparse
88
99from roboflow .config import API_URL , APP_URL , DEMO_KEYS , load_roboflow_api_key
1010from roboflow .core .project import Project
1313
1414__version__ = "1.0.0"
1515
16+
1617def check_key (api_key , model , notebook , num_retries = 0 ):
1718 if type (api_key ) is not str :
1819 raise RuntimeError (
@@ -127,8 +128,10 @@ def login(workspace=None, force=False):
127128 # we could eventually return the workspace object here
128129 # return Roboflow().workspace()
129130
131+
130132active_workspace = None
131133
134+
132135def initialize_roboflow ():
133136 global active_workspace
134137
@@ -138,29 +141,35 @@ def initialize_roboflow():
138141 )
139142
140143 if not os .path .isfile (conf_location ):
141- raise RuntimeError ("To use this method, you must first login - run roboflow.login()" )
144+ raise RuntimeError (
145+ "To use this method, you must first login - run roboflow.login()"
146+ )
142147 else :
143148 if active_workspace == None :
144149 active_workspace = Roboflow ().workspace ()
145150
146151 return active_workspace
147152
153+
148154def load_model (model_url ):
149155 operate_workspace = initialize_roboflow ()
150-
156+
151157 if "universe.roboflow.com" in model_url or "app.roboflow.com" in model_url :
152158 parsed_url = urlparse (model_url )
153159 path_parts = parsed_url .path .split ("/" )
154160 project = path_parts [2 ]
155- version = int (path_parts [- 1 ])
161+ version = int (path_parts [- 1 ])
156162 else :
157- raise ("Model URL must be from either app.roboflow.com or universe.roboflow.com" )
158-
163+ raise (
164+ "Model URL must be from either app.roboflow.com or universe.roboflow.com"
165+ )
166+
159167 project = operate_workspace .project (project )
160168 version = project .version (version )
161169 model = version .model
162170 return model
163171
172+
164173# continue distributing this object for back compatibility
165174class Roboflow :
166175 def __init__ (
0 commit comments