-
Notifications
You must be signed in to change notification settings - Fork 7
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
add base run functionality #253
Conversation
cleanlab_studio/studio/enrichment.py
Outdated
if options.get("tlm_options") | ||
else {}, | ||
) | ||
epr = EnrichmentResult.from_dict(response) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
full name var
cleanlab_studio/errors.py
Outdated
@@ -131,6 +135,10 @@ def __str__(self) -> str: | |||
return error_msg | |||
|
|||
|
|||
class EnrichmentProjectHandledError(CleansetHandledError): |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Don't subclass from CleansetHandledError
from cleanlab_studio.errors import EnrichmentProjectError, EnrichmentProjectHandledError | ||
from cleanlab_studio.internal.api import api | ||
from cleanlab_studio.internal.util import log_internal_error | ||
|
||
|
||
from tqdm import tqdm | ||
|
||
|
||
import itertools | ||
import time | ||
import traceback |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
nit: It looks like isort didn't run on this file. Do you have pre-commit installed for this repo?
constrain_outputs: Optional[List[str]] = None, | ||
extraction_pattern: Optional[str] = None, | ||
optimize_prompt: Optional[bool] = True, | ||
quality_preset: Optional[TLMQualityPreset] = "medium", |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
quality_preset: Optional[TLMQualityPreset] = "medium", | |
quality_preset: TLMQualityPreset = "medium", |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Why are we making quality_preset not optional here if it is optional in enrichment_preview()
method?
This PR adds Python API code for the base run functionality as well as wait until cleanset ready options.