Skip to content

Commit

Permalink
cmachak/standardize imports (#17)
Browse files Browse the repository at this point in the history
* organize imports

* bumps version

* fix __all__ in root module

* fix import
  • Loading branch information
xmachak authored Feb 2, 2024
1 parent 6752e19 commit 123383e
Show file tree
Hide file tree
Showing 4 changed files with 17 additions and 8 deletions.
10 changes: 10 additions & 0 deletions block_cascade/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,3 +9,13 @@
GcpResource,
)
from block_cascade.remote import remote # noqa: F401

__all__ = [
"DatabricksAutoscaleConfig",
"DatabricksResource",
"GcpAcceleratorConfig",
"GcpEnvironmentConfig",
"GcpMachineConfig",
"GcpResource",
"remote",
]
4 changes: 1 addition & 3 deletions block_cascade/config.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,9 +10,7 @@
from yaml import SafeLoader

from block_cascade.executors.databricks.resource import DatabricksResource
from block_cascade.executors.vertex.resource import (
GcpResource,
)
from block_cascade.executors.vertex.resource import GcpResource

logger = logging.getLogger(__name__)

Expand Down
9 changes: 5 additions & 4 deletions block_cascade/remote.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,13 +5,14 @@

import requests

from block_cascade import GcpResource
from block_cascade.config import find_default_configuration
from block_cascade.executors import DatabricksExecutor, LocalExecutor, VertexExecutor
from block_cascade.gcp import VMMetadataServerClient
from block_cascade.executors.databricks.resource import DatabricksResource
from block_cascade.executors.vertex import GcpEnvironmentConfig
from block_cascade.executors.databricks.executor import DatabricksExecutor
from block_cascade.executors.local.executor import LocalExecutor
from block_cascade.executors.vertex.executor import VertexExecutor
from block_cascade.executors.vertex.resource import GcpEnvironmentConfig, GcpResource
from block_cascade.executors.vertex.tune import Tune
from block_cascade.gcp import VMMetadataServerClient
from block_cascade.prefect import (
PrefectEnvironmentClient,
get_from_prefect_context,
Expand Down
2 changes: 1 addition & 1 deletion pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@ name = "block-cascade"
packages = [
{include = "block_cascade"}
]
version = "2.4.0"
version = "2.4.1"
description = "Library for model training in multi-cloud environment."
readme = "README.md"
authors = ["Block"]
Expand Down

0 comments on commit 123383e

Please sign in to comment.