38
38
from comet_ml .comet import format_url
39
39
from comet_ml .config import get_api_key , get_config
40
40
from comet_ml .connection import Reporting , get_comet_api_client , url_join
41
- from comet_ml .exceptions import CometRestApiException , NotFound
41
+ from comet_ml .exceptions import CometRestApiException
42
42
from comet_ml .offline import upload_single_offline_experiment
43
43
from mlflow .entities .run_tag import RunTag
44
44
from mlflow .entities .view_type import ViewType
@@ -528,17 +528,15 @@ def create_and_save_comet_project(self, exp, tag_name):
528
528
project_name = get_comet_project_name (self .store , exp .name )
529
529
530
530
# Check if the project exists already
531
- try :
532
- project = self .api_client .get_project (self .workspace , project_name )
533
- if not project :
534
- raise NotFound ("POST" , {})
535
- project_id = project ["projectId" ]
536
- except NotFound :
531
+ project = self .api_client .get_project (self .workspace , project_name )
532
+ if not project :
537
533
project = self .api_client .create_project (
538
534
self .workspace , project_name , public = False
539
535
)
540
536
541
537
project_id = project ["projectId" ]
538
+ else :
539
+ project_id = project ["projectId" ]
542
540
543
541
# Save the project id to the experiment tags
544
542
self .store .set_experiment_tag (exp .experiment_id , RunTag (tag_name , project_id ))
@@ -555,15 +553,12 @@ def get_or_create_comet_project(self, exp):
555
553
project_id = exp .tags [tag_name ]
556
554
557
555
# Check if the project exists
558
- try :
559
- project = self .api_client .get_project_by_id (project_id )
560
- if not project :
561
- raise NotFound ("POST" , {})
562
- return project ["projectName" ]
563
- except (NotFound ):
556
+ project = self .api_client .get_project_by_id (project_id )
557
+ if not project :
564
558
# A previous project ID has been saved but don't exists anymore (at
565
559
# least in this environment), recreate it
566
560
return self .create_and_save_comet_project (exp , tag_name )
561
+ return project ["projectName" ]
567
562
else :
568
563
return self .create_and_save_comet_project (exp , tag_name )
569
564
0 commit comments