Merge branch 'master' into restored_iceberg

doc/source/data/examples/pytorch_resnet_batch_prediction.ipynb

@@ -9,6 +9,8 @@
    "source": [
     "# Image Classification Batch Inference with PyTorch\n",
     "\n",
+    "[![try-anyscale-quickstart-image-classification-batch-inference-with-pytorch](../../_static/img/run-quickstart-anyscale.svg)](https://console.anyscale.com/v2/template-preview/ray-data-batch-image-classification?utm_source=ray_docs&utm_medium=docs&utm_campaign=image-classification-batch-inference-with-pytorch)\n",
+    "\n",
     "In this example, we will introduce how to use [Ray Data](data) for **large-scale batch inference with multiple GPU workers.**\n",
     "\n",
     "In particular, we will:\n",

python/ray/data/_internal/compute.py

@@ -88,7 +88,7 @@ def __init__(
                 computation and avoiding actor startup delays, but will also increase
                 queueing delay.
         """
-        if size:
+        if size is not None:
             if size < 1:
                 raise ValueError("size must be >= 1", size)
             if max_size is not None or min_size is not None:

src/ray/core_worker/context.cc

@@ -324,6 +324,7 @@ void WorkerContext::SetCurrentTask(const TaskSpecification &task_spec) {
 
 void WorkerContext::ResetCurrentTask() { GetThreadContext().ResetCurrentTask(); }
 
+/// NOTE: This method can't be used in fiber/async actor context.
 std::shared_ptr<const TaskSpecification> WorkerContext::GetCurrentTask() const {
   return GetThreadContext().GetCurrentTask();
 }
@@ -416,6 +417,7 @@ const ObjectID WorkerContext::GetGeneratorReturnId(
   return ObjectID::FromIndex(current_task_id, current_put_index);
 }
 
+/// NOTE: This method can't be used in fiber/async actor context.
 WorkerThreadContext &WorkerContext::GetThreadContext() const {
   if (thread_context_ == nullptr) {
     absl::ReaderMutexLock lock(&mutex_);

src/ray/core_worker/context.h

@@ -91,6 +91,7 @@ class WorkerContext {
 
   void ResetCurrentTask();
 
+  /// NOTE: This method can't be used in fiber/async actor context.
   std::shared_ptr<const TaskSpecification> GetCurrentTask() const;
 
   const ActorID &GetCurrentActorID() const ABSL_LOCKS_EXCLUDED(mutex_);
@@ -165,6 +166,7 @@ class WorkerContext {
   mutable absl::Mutex mutex_;
 
  private:
+  /// NOTE: This method can't be used in fiber/async actor context.
   WorkerThreadContext &GetThreadContext() const;
 
   /// Per-thread worker context.