Merge pull request #121 from manics/stdout

Ensure stdout works in Jupyter Notebooks

ome_model/experimental.py

@@ -6,11 +6,6 @@
 import uuid
 import xml.etree.ElementTree as ET
 
-if sys.version_info[0] > 2:
-    PYTHON = 3
-else:
-    PYTHON = 2
-
 OME_ATTRIBUTES = {
     'Creator': "ome_model/experimental.py",
     'UUID': "urn:uuid:%s" % uuid.uuid4(),
@@ -235,12 +230,14 @@ def create_companion(plates=[], images=[], out=None):
 
     # https://stackoverflow.com/a/48671499/56887
     kwargs = dict(encoding="UTF-8")
-    if PYTHON >= 3:
-        kwargs["xml_declaration"] = True
-        if not out:
+    kwargs["xml_declaration"] = True
+    if not out:
+        try:
             out = sys.stdout.buffer
-    elif not out:
-        out = sys.stdout
+        except AttributeError:
+            # ipykernel.iostream.OutStream (used by Jupyter Notebook) does not
+            # have a .buffer attribute
+            out = sys.stdout
     ET.ElementTree(root).write(out, **kwargs)