Commit graph

5226 commits

Author SHA1 Message Date
Edward Oakes
9cf19b67cc
[serve] Remove log poll client from replicas (#19145)
In general, broadcasting changes to the replicas via the LongPollClient is hard to reason about (it circumvents our versioning semantics as there's no rolling update). Ideally we would only be using the LongPollClient to broadcast replica membership and nothing else.
2021-10-08 12:32:42 -05:00
Edward Oakes
86d1a5bfc6
[serve] Catch ConnectionError during shutdown in LongPollClient (#19224) 2021-10-08 12:31:35 -05:00
Edward Oakes
93bcea7bdd
[serve] Clean up kv store file, skip on windows (#19194) 2021-10-08 12:30:48 -05:00
Kai Fricke
44b0b6eb20
[tune/wip] Exclude trial checkpoints in experiment sync (#19185) 2021-10-08 18:26:03 +01:00
Kai Fricke
e5e1ba93d9
[tune] Use queue to display JupyterNotebookReporter updates in Ray client (#19137) 2021-10-08 18:23:20 +01:00
Antoni Baum
c7d6f838f6
[tune] Optional forcible trial cleanup, return default autofilled metrics even if Trainable doesn't report at least once (#19144) 2021-10-08 18:16:26 +01:00
Eric Liang
8beabb283b
Force disable placement_group for all dataset tasks (#19208) 2021-10-08 10:16:09 -07:00
Kai Fricke
f1606acc2b
[tune] Fix durable(str) name for class trainables, preventing trial recovery (#19223) 2021-10-08 17:32:05 +01:00
architkulkarni
1aab892623
[Runtime Env] add excludes to known fields for runtime env (#19206) 2021-10-07 22:47:49 -07:00
Eric Liang
8dded14798
Refactor LazyBlockList to simplify union of lists (#19214) 2021-10-07 22:07:52 -07:00
SangBin Cho
afaee05e1e
[Placement Group] Fix placement group removal leak (#19138) 2021-10-07 22:04:12 -07:00
Simon Mo
46e80348ad
[Serve] Make long poll wait for non-existent keys (#19205) 2021-10-07 19:10:22 -07:00
Kai Fricke
8d89e2d546
[tune] Prevent errors with retained trainables in global registry (#19184)
This PR fixes #19183 by introducing three improvements:

String trainables are prefixed with Durable, e.g. DurablePPO
Durable trainables cannot be wrapped twice with tune.durable()
MRO resolution in _WrappedDurableTrainables indicates we already have a DurableTrainable - thus we catch this with a try/except block
2021-10-07 17:17:01 -07:00
Edward Oakes
454163912f
Revert "[serve] Delete kv store local path after unit tests (#19165)" (#19188)
This reverts commit b90af4dae5.
2021-10-07 14:26:18 -05:00
Edward Oakes
1fa81673bd
[runtime_env] Clean up validation logic (#18984)
Splits the runtime_env parsing/validation and overriding into two separate codepaths. Adds unit testing for both.
2021-10-07 14:24:41 -05:00
Kai Fricke
45aad4ee9a
[tune] Add resume="AUTO" and enhance resume error messages (#19181) 2021-10-07 19:00:56 +01:00
Stephanie Wang
940f84cedb
[core] Remove unused plasma promotion path (#19122)
* remove unused

* lint

* lint

* lint
2021-10-07 10:55:50 -07:00
SangBin Cho
0ef0d9a77d
Revert "[core] Assign tasks to the first available worker (#18167)" (#19180)
This reverts commit 545db13800.
2021-10-07 10:38:37 -07:00
Antoni Baum
f1587c06fd
[tune] Ensure loc in progress reporter is filled (#19182) 2021-10-07 15:43:49 +01:00
Edward Oakes
b90af4dae5
[serve] Delete kv store local path after unit tests (#19165) 2021-10-07 08:55:22 -05:00
Kai Fricke
a8cf8c648c
[tune] track and print elapsed time in reporters (#19139) 2021-10-07 10:56:17 +01:00
Avnish Narayan
bbc64a7c3d
[RLlib] Pin Gym to 0.19 (#19170)
Gym appears to have cut a release, 0.21.
It isn't clear what changes were made
between 0.19/0.20 and 0.21, as there is
no change log available for the 0.21 release,
so for now we'll pin gym to 0.19 until we
can fully understand the breaking changes
in gym 0.21. I suspect some things have
just been removed from the regular gym installation
that rllib has previously relied on. Will address
later.
2021-10-07 07:59:02 +02:00
mwtian
fe413c3c5e
[Client] disable auto init for get_runtime_context() (#19127) 2021-10-06 20:20:47 -07:00
Eric Liang
86cbe3e833
[data] Add support for repeating and re-windowing a DatasetPipeline (#19091) 2021-10-06 20:13:43 -07:00
Edward Oakes
0f915820e1
[serve] Rename backend_worker -> replica (#19150) 2021-10-06 16:39:17 -05:00
Chris K. W
d1517c33ab
[client] deflake test_object_ref_cleanup (#19153) 2021-10-06 14:06:43 -07:00
Kai Fricke
9f77cd8d28
[tune] Deflake PBT Async test (#19135) 2021-10-06 12:24:22 -07:00
Edward Oakes
9316a9977f
[serve] Support kwargs to deployment constructor (#19023) 2021-10-06 14:16:23 -05:00
Frank Luan
77d0a08c38
[docker] Fix missing space in docker.py warning (#19128) 2021-10-06 12:09:26 -07:00
Ian Rodney
8cab8d3ae9
[Datasets] Clean Up docs around pipelining -> windowing rename (#19142) 2021-10-06 11:07:55 -07:00
Chris K. W
db1105fa83
[client] Skip test_valid_actor_state tests on windows (#19114)
* skip test_wrapped_actor_creation on windows

* rerun windows ci

* mark test_valid_actor_state_2 as flaky

* mark test_valid_actor_state

* rerun
2021-10-06 09:17:59 -07:00
Amog Kamsetty
db0483a29a
[SGD] SGD Namespace Consistency (#19048)
* wip

* update

* add callbacks

* fix

* fix

* update

* add

* address comments
2021-10-05 15:56:42 -07:00
Matti Picus
63dd22c7c2
add msvcp140.dll to the wheel on windows (#19062)
* add msvcp140.dll to the wheel on windows

* fixes from review

* be more verbose

* Update setup.py

Co-authored-by: Philipp Moritz <pcmoritz@gmail.com>
2021-10-05 15:12:46 -07:00
Stephanie Wang
545db13800
[core] Assign tasks to the first available worker (#18167)
* Convert worker pool to queue

* Start up to backlog size more workers

* fixes

* Prestart workers according to num available CPUs

* lint

* x

* Update src/ray/raylet/worker_pool.h

Co-authored-by: Eric Liang <ekhliang@gmail.com>

* Update src/ray/raylet/worker_pool.h

Co-authored-by: Eric Liang <ekhliang@gmail.com>

* dedicated workers

* Fix tests

* x

* fix

* asan

* asan

* Workers can only exec tasks with same job ID

* size_t for runtime env hash, fix unit tests

* include job ID in runtime env hash, remove from worker registration msg

* x

* conflict

* debug

* Schedule and dispatch periodically, skip if no new tasks

* Update src/ray/common/task/task_spec.h

Co-authored-by: Eric Liang <ekhliang@gmail.com>

* Update src/ray/raylet/scheduling/cluster_task_manager.h

Co-authored-by: Eric Liang <ekhliang@gmail.com>

* Update src/ray/raylet/worker_pool.h

Co-authored-by: Eric Liang <ekhliang@gmail.com>

Co-authored-by: Eric Liang <ekhliang@gmail.com>
2021-10-05 13:45:50 -07:00
Yi Cheng
ecf7b86585
[workflow] Avoid running workflow step multiple times. (#19090)
When workflow recover, it'll try to reconstruct the DAG. However, it's step scoped, which means if a workflow is passed to multiple steps, it'll be executed multiple times which breaks the exactly-once semantic.

For ObjectRef it's ok since it'll be cached with serialization context, but we also need a similar thing for Workflow input.

This logic is put in workflow layer instead of serialization layer because it's dedupe on app layer.

Issue #18997 has race conditions, and it's also related to this one. The reason is that multiple steps will try to issue writes to virtual actors at the same time which is not allowed right now and can lead to race condition.
2021-10-05 13:43:27 -07:00
Kai Fricke
957f9e9d99
[client] Undo PySpark's monkey patching of namedtuples for PickleStub (#19034) 2021-10-05 10:43:50 -07:00
SangBin Cho
83cb992d5b
Revert pull retry (#19068)
* Revert "[Object manager] fix comments"

This reverts commit 56debfc063.

* Revert "[Object manager] don't abort entire pull request on race condition in concurrent chunk receive (#18955)"

This reverts commit d12e35ce53.

* Fix a lint issue
2021-10-04 11:20:43 -07:00
SangBin Cho
7fcf1bf57e
[Dashboard] Refine the dashboard restart logic. (#18973)
* in progress

* Refine the dashboard agent retry logic

* refine

* done

* lint
2021-10-04 05:01:51 -07:00
Jiajun Yao
7588bfd315
[Lint] Add flake8-bugbear (#19053)
* Add flake8-bugbear

* Add flake8-bugbear
2021-10-03 23:24:11 -07:00
Siyuan (Ryans) Zhuang
28d905dcb0
[Workflow] Move arguments into workflow step context (#19003)
* refactor

* improve documentation

* fix comments

* Use dataclass for workflow context

* update docs
2021-10-01 23:48:57 -07:00
Eric Liang
032a420ee6
Rename Dataset.pipeline to Dataset.window (#19050) 2021-10-01 19:55:29 -07:00
Kai Fricke
3dc176c42e
[ci/tune] Add SGD and Tune GPU pipeline step to CI (#18469)
* [ci/tune] Add Tune GPU pipeline step to CI

* cont.

* add sgd gpu tests

* format yaml, fix imports

* install horovod; fix line wrapping

* set GPU per worker to 0.5

* fix import

* move test to 4gpu machine

* fix lint

* lint

* set visible devices

* pull in tf gpu fix

* Fix Tune GPU pipeline step

* nit

* Disable GPU tests until we have some

* Re-add empty rllib tests

Co-authored-by: Matthew Deng <matthew.j.deng@gmail.com>
2021-10-01 18:34:05 -07:00
Simon Mo
9b2a368c8c
[Runtime Env] Implement basic runtime env plugin mechanism (#19044) 2021-10-01 17:22:54 -07:00
Edward Oakes
cac6f9d75c
skip test on windows (#19047) 2021-10-01 15:56:37 -07:00
Ian Rodney
a4ebe2697c
[Autoscaler] Improve assert_called (#19036)
* improvements

* fix invocations

* improve not_has_call
2021-10-01 14:08:31 -07:00
Clark Zinzow
d22f838795
[Datasets] Delineate between ref and raw APIs for the Pandas/Arrow integrations. (#18992) 2021-10-01 13:08:25 -07:00
Frank Luan
f885060efa
Disable distributed sort test on Windows (#19041)
* [WIP] Sorting benchmark

* Separate num_mappers and num_reducers

* Add tests

* Fix tests

* Tracing

* Separate num_mappers and num_reducers

* Two-stage reduce

* Back pressure to avoid excessive spilling

* Make merger_concurrency an option

* Fix tests

* Tweaks

* Remote writers

* Format

* WIP

* Address comments

* Fix tests and address comments

* Lint

* Fix mount points for testing

* Simplify code path

* Address comments

* Disable distributed sort test on Windows
2021-10-01 12:17:28 -07:00
mwtian
56debfc063
[Object manager] fix comments 2021-10-01 11:42:07 -07:00
Stephanie Wang
c052395f4e
[core] Remove "plasma promotion" for serialized ObjectRefs 2021-10-01 10:39:55 -07:00
architkulkarni
b0a5564f4e
[Serve] Integrate metrics with minimal autoscaling algorithm and add e2e test (#18793) 2021-10-01 10:21:12 -07:00