2023-03-20 20:28:43 +00:00
|
|
|
package resources
|
|
|
|
|
2023-09-04 09:55:01 +00:00
|
|
|
import (
|
|
|
|
"github.com/databricks/cli/bundle/config/paths"
|
2023-10-16 06:56:06 +00:00
|
|
|
"github.com/databricks/databricks-sdk-go/marshal"
|
2023-09-04 09:55:01 +00:00
|
|
|
"github.com/databricks/databricks-sdk-go/service/ml"
|
|
|
|
)
|
2023-03-20 20:28:43 +00:00
|
|
|
|
|
|
|
type MlflowModel struct {
|
2023-03-21 09:58:16 +00:00
|
|
|
Permissions []Permission `json:"permissions,omitempty"`
|
|
|
|
|
2023-09-04 09:55:01 +00:00
|
|
|
paths.Paths
|
2023-04-12 14:17:13 +00:00
|
|
|
|
2023-04-21 08:30:20 +00:00
|
|
|
*ml.Model
|
2023-03-20 20:28:43 +00:00
|
|
|
}
|
2023-10-16 06:56:06 +00:00
|
|
|
|
|
|
|
func (s *MlflowModel) UnmarshalJSON(b []byte) error {
|
|
|
|
return marshal.Unmarshal(b, s)
|
|
|
|
}
|
|
|
|
|
|
|
|
func (s MlflowModel) MarshalJSON() ([]byte, error) {
|
|
|
|
return marshal.Marshal(s)
|
|
|
|
}
|