ui: import sys on write_json

iterative · Sep 29, 2021 · 3f5f983 · 3f5f983
1 parent fed9f08
commit 3f5f983
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/dvc/ui/__init__.py b/dvc/ui/__init__.py
@@ -87,7 +87,7 @@ def write_json(
         data: Any,
         indent: int = 2,
     ) -> None:
-        if sys.stdout.isatty():
+        if self.isatty():
             from rich.json import JSON
 
             j = JSON.from_data(data, indent=indent)
@@ -237,6 +237,11 @@ def table(
     def status(self, status: str, **kwargs: Any) -> "Status":
         return self.error_console.status(status, **kwargs)
 
+    def isatty(self) -> bool:
+        import sys
+
+        return sys.stdout.isatty()
+
 
 ui = Console()