aztk.models package

aztk.models.models module

class aztk.models.models.FileShare(storage_account_name: str = None, storage_account_key: str = None, file_share_path: str = None, mount_path: str = None)[source]

Bases: object

class aztk.models.models.File(name: str, payload: _io.StringIO)[source]

Bases: object

class aztk.models.models.CustomScript(name: str = None, script=None, run_on=None)[source]

Bases: object

class aztk.models.models.UserConfiguration(username: str, ssh_key: str = None, password: str = None)[source]

Bases: aztk.internal.configuration_base.ConfigurationBase

merge(other)[source]
validate()[source]
class aztk.models.models.ClusterConfiguration(toolkit: aztk.models.toolkit.Toolkit = None, custom_scripts: List[aztk.models.models.CustomScript] = None, file_shares: List[aztk.models.models.FileShare] = None, cluster_id: str = None, vm_count=0, vm_low_pri_count=0, vm_size=None, subnet_id=None, plugins: List[aztk.models.plugins.plugin_configuration.PluginConfiguration] = None, user_configuration: aztk.models.models.UserConfiguration = None)[source]

Bases: aztk.internal.configuration_base.ConfigurationBase

Cluster configuration model

Parameters:toolkit
merge(other)[source]

Merge other cluster config into this one. :params other: ClusterConfiguration

mixed_mode() → bool[source]
gpu_enabled()[source]
get_docker_repo()[source]
validate() → bool[source]

Validate the config at its current state. Raises: Error if invalid

class aztk.models.models.RemoteLogin(ip_address, port)[source]

Bases: object

class aztk.models.models.ServicePrincipalConfiguration(tenant_id: str = None, client_id: str = None, credential: str = None, batch_account_resource_id: str = None, storage_account_resource_id: str = None)[source]

Bases: aztk.internal.configuration_base.ConfigurationBase

Container class for AAD authentication

validate() → bool[source]

Validate the config at its current state. Raises: Error if invalid

class aztk.models.models.SharedKeyConfiguration(batch_account_name: str = None, batch_account_key: str = None, batch_service_url: str = None, storage_account_name: str = None, storage_account_key: str = None, storage_account_suffix: str = None)[source]

Bases: aztk.internal.configuration_base.ConfigurationBase

Container class for shared key authentication

validate() → bool[source]

Validate the config at its current state. Raises: Error if invalid

class aztk.models.models.DockerConfiguration(endpoint=None, username=None, password=None)[source]

Bases: aztk.internal.configuration_base.ConfigurationBase

validate()[source]
class aztk.models.models.SecretsConfiguration(service_principal=None, shared_key=None, docker=None, ssh_pub_key=None, ssh_priv_key=None)[source]

Bases: aztk.internal.configuration_base.ConfigurationBase

validate()[source]
is_aad()[source]
class aztk.models.models.VmImage(publisher, offer, sku)[source]

Bases: object

class aztk.models.models.Cluster(pool: azure.batch.models.cloud_pool.CloudPool, nodes: azure.batch.models.compute_node_paged.ComputeNodePaged = None)[source]

Bases: object

class aztk.models.models.SSHLog(output, node_id)[source]

Bases: object

class aztk.models.models.Software[source]

Bases: object

Enum with list of available softwares

spark = 'spark'