flyte 0.2.0b25__py3-none-any.whl → 0.2.0b27__py3-none-any.whl

flyte/remote/_logs.py

@@ -10,8 +10,10 @@ from rich.panel import Panel
 from rich.text import Text
 
 from flyte._initialize import ensure_client, get_client
+from flyte._logging import logger
 from flyte._protos.logs.dataplane import payload_pb2
 from flyte._protos.workflow import run_definition_pb2, run_logs_service_pb2
+from flyte._tools import ipython_check, ipywidgets_check
 from flyte.errors import LogsNotYetAvailableError
 from flyte.syncify import syncify
 
@@ -158,6 +160,12 @@ class Logs:
         """
         if attempt < 1:
             raise ValueError("Attempt number must be greater than 0.")
+
+        if ipython_check():
+            if not ipywidgets_check():
+                logger.warning("IPython widgets is not available, defaulting to console output.")
+                raw = True
+
         if raw:
             console = Console()
             async for line in cls.tail.aio(action_id=action_id, attempt=attempt):

flyte/remote/_project.py

@@ -6,10 +6,11 @@ from typing import AsyncIterator, Iterator, Literal, Tuple, Union
 import rich.repr
 from flyteidl.admin import common_pb2, project_pb2
 
-from flyte._initialize import ensure_client, get_client, get_common_config
+from flyte._initialize import ensure_client, get_client
 from flyte.syncify import syncify
 
 
+# TODO Add support for orgs again
 @dataclass
 class Project:
     """
@@ -32,7 +33,7 @@ class Project:
         resp = await service.GetProject(
             project_pb2.ProjectGetRequest(
                 id=name,
-                org=org,
+                # org=org,
             )
         )
         return cls(resp)
@@ -57,7 +58,7 @@ class Project:
         sort_pb2 = common_pb2.Sort(
             key=sort_by[0], direction=common_pb2.Sort.ASCENDING if sort_by[1] == "asc" else common_pb2.Sort.DESCENDING
         )
-        org = get_common_config().org
+        # org = get_common_config().org
         while True:
             resp = await get_client().project_domain_service.ListProjects(  # type: ignore
                 project_pb2.ProjectListRequest(
@@ -65,7 +66,7 @@ class Project:
                     token=token,
                     filters=filters,
                     sort_by=sort_pb2,
-                    org=org,
+                    # org=org,
                 )
             )
             token = resp.token