Skip to content

Commit

Permalink
make linter happy
Browse files Browse the repository at this point in the history
Signed-off-by: Niels Bantilan <[email protected]>
  • Loading branch information
cosmicBboy committed Oct 7, 2024
1 parent 651a3b4 commit c1b473d
Showing 1 changed file with 3 additions and 0 deletions.
3 changes: 3 additions & 0 deletions examples/neptune_plugin/neptune_plugin/neptune_example.py
Original file line number Diff line number Diff line change
Expand Up @@ -62,6 +62,7 @@
# %% [markdown]
# First, we use a task to download the dataset and cache the data in Flyte:


# %%
@task(
container_image=image,
Expand All @@ -85,6 +86,7 @@ def get_dataset() -> Tuple[np.ndarray, np.ndarray]:
# in Neptune's integrations. In this example, we pass the `Run` object into Neptune's
# XGBoost callback.


# %%
@task(
container_image=image,
Expand Down Expand Up @@ -133,6 +135,7 @@ def train_model(max_depth: int, X: np.ndarray, y: np.ndarray):
# With Flyte's dynamic workflows, we can scale up multiple training jobs with different
# `max_depths`:


# %%
@dynamic(container_image=image)
def train_multiple_models(max_depths: List[int], X: np.ndarray, y: np.ndarray):
Expand Down

0 comments on commit c1b473d

Please sign in to comment.