Skip to content

Commit

Permalink
Formatting
Browse files Browse the repository at this point in the history
  • Loading branch information
sei-aderr committed Jan 14, 2025
1 parent 71f3611 commit 856fb0c
Showing 1 changed file with 2 additions and 6 deletions.
8 changes: 2 additions & 6 deletions mlte/store/catalog/sample_catalog.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,12 +6,11 @@

from __future__ import annotations

import importlib.resources
import json
import os
from typing import Optional

import mlte.store.catalog.sample as sample_entries
from mlte._private.reflection import get_json_resources
from mlte.catalog.model import CatalogEntry
from mlte.store.base import StoreType, StoreURI
from mlte.store.catalog.factory import create_catalog_store
Expand All @@ -20,7 +19,6 @@
CatalogStoreSession,
ManagedCatalogSession,
)
from mlte._private.reflection import get_json_resources


class SampleCatalog:
Expand Down Expand Up @@ -86,9 +84,7 @@ def _populate_catalog(catalog_session: CatalogStoreSession) -> None:
num_entries = 0
for json_data in get_json_resources(sample_entries):
entry = CatalogEntry(**json_data)
entry.header.catalog_id = (
SampleCatalog.SAMPLE_CATALOG_ID
)
entry.header.catalog_id = SampleCatalog.SAMPLE_CATALOG_ID
catalog_session.entry_mapper.create(entry)
num_entries += 1
print(f"Loaded {num_entries} entries for sample catalog.")

0 comments on commit 856fb0c

Please sign in to comment.