Skip to content

Commit

Permalink
chore: make constants upper case
Browse files Browse the repository at this point in the history
  • Loading branch information
burtenshaw committed Jul 10, 2024
1 parent fbc4c3a commit 13363a6
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions argilla/src/argilla/datasets/_export/_hub.py
Original file line number Diff line number Diff line change
Expand Up @@ -30,8 +30,8 @@


class HubImportExportMixin:
_default_dataset_repo_dir_name = ".argilla"
_default_configuration_files = ["settings.json", "dataset.json"]
_DEFAULT_DATASET_REPO_DIR = ".argilla"
_DEFAULT_CONFIGURATION_FILES = ["settings.json", "dataset.json"]

def to_hub(
self: "Dataset",
Expand Down Expand Up @@ -70,7 +70,7 @@ def to_hub(
repo_id=repo_id,
repo_type="dataset",
token=kwargs.get("token"),
path_in_repo=self._default_dataset_repo_dir_name,
path_in_repo=self._DEFAULT_DATASET_REPO_DIR,
)

if generate_card:
Expand Down Expand Up @@ -119,7 +119,7 @@ def from_hub(
from huggingface_hub import hf_hub_download

with TemporaryDirectory() as tmpdirname:
for filename in cls._default_configuration_files:
for filename in cls._DEFAULT_CONFIGURATION_FILES:
hf_hub_download(
repo_id=repo_id,
filename=filename,
Expand Down

0 comments on commit 13363a6

Please sign in to comment.