livebook-dev/livebook
1
1
Fork 0
mirror of https://github.com/livebook-dev/livebook.git synced 2024-09-20 01:55:56 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add smart cell definition FLAME runner cell with k8s backend (#2762)

Browse source 
Co-authored-by: Jonatan Kłosko <jonatanklosko@gmail.com>
This commit is contained in:
Michael Ruoss 2024-09-05 18:33:55 +02:00 committed by GitHub
parent 30c6fbbe12
commit c060efba20
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 16 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
lib/livebook/runtime/definitions.ex
View file

@ -57,7 +57,12 @@ defmodule Livebook.Runtime.Definitions do
kino_flame = %{
name: "kino_flame",
dependency: %{dep: {:kino_flame, "~> 0.1.0"}, config: []}
dependency: %{dep: {:kino_flame, "~> 0.1.3"}, config: []}
}
flame_k8s_backend = %{
name: "flame_k8s_backend",
dependency: %{dep: {:flame_k8s_backend, "~> 0.5"}, config: []}
}
jason = %{
@ -75,6 +80,11 @@ defmodule Livebook.Runtime.Definitions do
dependency: %{dep: {:xlsx_reader, "~> 0.8.5"}, config: []}
}
yaml_elixir = %{
name: "yaml_elixir",
dependency: %{dep: {:yaml_elixir, "~> 2.0"}, config: []}
}
windows? = match?({:win32, _}, :os.type())
nx_backend_package = if(windows?, do: torchx, else: exla)
@ -213,8 +223,12 @@ defmodule Livebook.Runtime.Definitions do
name: "FLAME runner cell",
requirement_presets: [
%{
name: "Default",
name: "Fly",
packages: [kino_flame]
},
%{
name: "Kubernetes",
packages: [kino_flame, flame_k8s_backend, yaml_elixir]
}
]
}