[packaging] update setup to be GH action compatible.

outerbounds · Sep 30, 2024 · dfc19fc · dfc19fc
1 parent 9c90676
commit dfc19fc
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -35,7 +35,7 @@ def start(self):
  self.next(self.train, num_parallel=NUM_NODES)
 
 @metaflow_ray
-@batch(**RESOURCES)
+@batch(**RESOURCES) # You can even set @kubernetes 
 @step
 def train(self):
  # Your step's training code here
@@ -44,7 +44,7 @@ def train(self):
 3. Initialize Ray within Your Step
 ```python
 @metaflow_ray
-@batch(**RESOURCES)
+@batch(**RESOURCES) # You can even set @kubernetes 
 @step
 def train(self):
  import ray

diff --git a/setup.py b/setup.py
@@ -2,11 +2,16 @@
 
 version = "0.1.0"
 
+def get_long_description() -> str:
+ with open("README.md") as fh:
+ return fh.read()
+
 setup(
  name="metaflow-ray",
  version=version,
  description="An EXPERIMENTAL Ray decorator for Metaflow",
  author="Riley Hun",
+ long_description=get_long_description(),
  author_email="[email protected]",
  packages=find_namespace_packages(include=["metaflow_extensions.*"]),
  py_modules=[