Skip to content

Commit

Permalink
typo
Browse files Browse the repository at this point in the history
  • Loading branch information
matt-bernstein committed Nov 26, 2024
1 parent 77bf44b commit 54148fe
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions adala/skills/collection/label_studio.py
Original file line number Diff line number Diff line change
Expand Up @@ -41,10 +41,10 @@ class LabelStudioSkill(TransformSkill):
@cached_property
def ner_tags(self) -> Iterator[ControlTag]:
# check if the input config has NER tag (<Labels> + <Text>), and return its `from_name` and `to_name`
interface = LabelInterface(self.label_config)
control_tag_names = self.allowed_control_tags or list(
interface._controls.keys()
)
interface = LabelInterface(self.label_config)
for tag_name in control_tag_names:
tag = interface.get_control(tag_name)
if (
Expand All @@ -56,8 +56,8 @@ def ner_tags(self) -> Iterator[ControlTag]:
@cached_property
def image_tags(self) -> Iterator[ObjectTag]:
# check if any image tags are used as input variables
object_tag_names = self.allowed_object_tags or list(interface._objects.keys())
interface = LabelInterface(self.label_config)
object_tag_names = self.allowed_object_tags or list(interface._objects.keys())
for tag_name in object_tag_names:
tag = interface.get_object(tag_name)
if tag.tag.lower() == "image":
Expand Down

0 comments on commit 54148fe

Please sign in to comment.