fix: add comments from alvarobartt on pr argilla-io#3969

plaguss · Oct 18, 2023 · cd16b9e · cd16b9e
1 parent 2ca704c
commit cd16b9e
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -137,6 +137,9 @@ version = { attr = "argilla.__version__" }
 "argilla.client.feedback.integrations.huggingface.card" = [
     "argilla_template.md",
 ]
+"argilla.client.feedback.integrations.huggingface.model_card" = [
+    "argilla_model_template.md",
+]
 
 [tool.pytest.ini_options]
 log_format = "%(asctime)s %(name)s %(levelname)s %(message)s"

diff --git a/src/argilla/client/feedback/integrations/huggingface/card/__init__.py b/src/argilla/client/feedback/integrations/huggingface/card/__init__.py
@@ -15,7 +15,4 @@
 from ._dataset_card import ArgillaDatasetCard
 from ._parser import size_categories_parser
 
-__all__ = [
-    "ArgillaDatasetCard",
-    "size_categories_parser",
-]
+__all__ = ["ArgillaDatasetCard", "size_categories_parser"]
diff --git a/src/argilla/client/feedback/integrations/huggingface/model_card/__init__.py b/src/argilla/client/feedback/integrations/huggingface/model_card/__init__.py
@@ -12,7 +12,7 @@
 #  See the License for the specific language governing permissions and
 #  limitations under the License.
 
-from .model_card import (
+from argilla.client.feedback.integrations.huggingface.model_card.model_card import (
     ArgillaModelCard,
     FrameworkCardData,
     OpenAIModelCardData,