Add Fly.io as a provider

ci/scripts/test_imports.sh

@@ -19,3 +19,4 @@ test_import "aws"               "import dask_cloudprovider.aws"
 test_import "azure"             "import dask_cloudprovider.azure"
 test_import "digitalocean"      "import dask_cloudprovider.digitalocean"
 test_import "gcp"               "import dask_cloudprovider.gcp"
+test_import "fly"               "import dask_cloudprovider.fly"

dask_cloudprovider/__init__.py

@@ -42,3 +42,9 @@ def __getattr__(name):
             "DigitalOcean cluster managers must be imported from the digitalocean subpackage. "
             f"Please import dask_cloudprovider.digitalocean.{name}"
         )
+
+    if name in ["FlyMachineCluster"]:
+        raise ImportError(
+            "Fly.io cluster managers must be imported from the fly subpackage. "
+            f"Please import dask_cloudprovider.fly.{name}"
+        )

dask_cloudprovider/cloudprovider.yaml

@@ -117,3 +117,13 @@ cloudprovider:
     image: "ubuntu-20.04" # Operating System image to use
     docker_image: "daskdev/dask:latest" # docker image to use
     bootstrap: true # It is assumed that the OS image does not have Docker and needs bootstrapping. Set this to false if using a custom image with Docker already installed.
+
+  fly:
+    token: null # API token for interacting with the Fly API
+    region: "lax" # Region to launch Droplets in
+    vm_size: "shared-cpu-1x" # Droplet size to launch, default is 1GB RAM, 1 vCPU
+    image: "daskdev/dask:latest" # Operating System image to use
+    memory_mb: 1024 # Memory in MB to use for the scheduler and workers
+    cpus: 1 # Number of CPUs to use for the scheduler and workers
+    app_name: null # Name of Fly app to use. If it is blank, a random name will be generated.
+    org_slug: "personal" # Organization slug to use. If it is blank, the personal organization will be used.

dask_cloudprovider/fly/__init__.py

@@ -0,0 +1,6 @@
+from .machine import (
+    FlyMachine,
+    FlyMachineWorker,
+    FlyMachineScheduler,
+    FlyMachineCluster,
+)