Documentation
¶
Index ¶
Constants ¶
This section is empty.
Variables ¶
This section is empty.
Functions ¶
This section is empty.
Types ¶
type MlFlowClient ¶
type MlFlowClient struct {
// contains filtered or unexported fields
}
func NewMlFlowRegistry ¶
func NewMlFlowRegistry(mlflowClient *client.DatabricksClient) (MlFlowClient, error)
func (*MlFlowClient) PullModelByName ¶
func (mlfr *MlFlowClient) PullModelByName( ctx context.Context, modelName string, modelVersion string, destSrc string, ) (string, error)
Pull latest modelVersion if `modelVersion` is nil
type MlflowProvider ¶
type MlflowProvider struct {
// contains filtered or unexported fields
}
MlflowProvider implements the modelprovider.Provider interface for Mlflow
func (*MlflowProvider) CheckAuth ¶
func (p *MlflowProvider) CheckAuth() error
CheckAuth verifies that the user is authenticated with MlFlow
func (*MlflowProvider) DownloadModel ¶
func (p *MlflowProvider) DownloadModel( ctx context.Context, modelURL, destDir string, ) (string, error)
DownloadModel downloads a model from ModelScope using the modelscope CLI
func (*MlflowProvider) Name ¶
func (p *MlflowProvider) Name() string
Name returns the name of this provider
func (*MlflowProvider) SupportsURL ¶
func (p *MlflowProvider) SupportsURL(url string) bool
SupportsURL checks if this provider can handle the given URL Supports experiments url (runs), model-registry(models) For short-form repo identifiers (owner/repo), users must explicitly specify --provider mlflow
type S3StorageBackend ¶
type S3StorageBackend struct {
// contains filtered or unexported fields
}
func (*S3StorageBackend) DownloadModel ¶
Source Files
Click to show internal directories.
Click to hide internal directories.