add import to init (#621)

* add import to init

* update tasks to fetch_tasks

* include tasks for backwards compat

src/tiledb/cloud/__init__.py

@@ -34,6 +34,9 @@
 from .notebook import upload_notebook_contents
 from .notebook import upload_notebook_from_file
 from .rest_api import models
+from .tasks import fetch_results
+from .tasks import fetch_results_pandas
+from .tasks import fetch_tasks
 from .tasks import last_sql_task
 from .tasks import last_udf_task
 from .tasks import task
@@ -86,5 +89,8 @@
  "last_sql_task",
  "last_udf_task",
  "task",
+ "fetch_tasks",
+ "fetch_results",
+ "fetch_results_pandas",
  "TileDBCloudError",
 )

src/tiledb/cloud/tasks.py

@@ -1,6 +1,6 @@
 import datetime
 import uuid
-from typing import TYPE_CHECKING, Any, Optional
+from typing import TYPE_CHECKING, Any, Dict, Optional, Sequence
 
 from tiledb.cloud import array
 from tiledb.cloud import client
@@ -37,18 +37,18 @@ def task(id, async_req=False):
  raise tiledb_cloud_error.maybe_wrap(exc) from None
 
 
-def tasks(
+def fetch_tasks(
  namespace=None,
  array=None,
  start=None,
- end=datetime.datetime.utcnow(),
+ end=datetime.datetime.now(datetime.timezone.utc),
  status=None,
  page=None,
  per_page=None,
  async_req=False,
-):
+) -> Dict[str, Sequence[Dict[str, Any]]]:
  """
- Fetch all tasks a user has access too
+ Fetch all tasks a user has access too.
  :param str namespace: optional filter by namespace
  :param str array: optional limit tasks to specific array
  :param datetime start: optional start time for listing of tasks,
@@ -59,8 +59,9 @@ def tasks(
  :param int page: optional page for pagenating results
  :param int per_page: optional records to return per page
  :param async_req: return future instead of results for async support
- :return:
+ :return: Mapping of task data organized by task type.
  """
+
  api_instance = client.build(rest_api.TasksApi)
 
  if end is not None:
@@ -108,6 +109,10 @@ def tasks(
  raise tiledb_cloud_error.maybe_wrap(exc) from None
 
 
+# maintain backwards compatibility for tiledb.cloud.tasks.tasks
+tasks = fetch_tasks
+
+
 def last_sql_task():
  """
  Fetch the last run sql array task

tests/test_basic.py

@@ -49,7 +49,7 @@ def test_array_activity(self):
  array.array_activity("tiledb://TileDB-Inc/quickstart_sparse")
 
  def test_tasks(self):
- self.assertIsNotNone(tasks.tasks(page=1, per_page=100))
+ self.assertIsNotNone(tasks.fetch_tasks(page=1, per_page=100))
 
  def test_list_arrays(self):
  self.assertIsNotNone(client.list_arrays().arrays)