Update base.py

talagayev · Aug 6, 2024 · 6a9fa07 · 6a9fa07
1 parent 17f863e
commit 6a9fa07
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/package/MDAnalysis/coordinates/base.py b/package/MDAnalysis/coordinates/base.py
@@ -1663,7 +1663,15 @@ class SingleFrameReaderBase(ProtoReader):
     def __init__(self, filename, convert_units=True, n_atoms=None, **kwargs):
         super(SingleFrameReaderBase, self).__init__()
 
-        self.filename = filename
+        if isinstance(filename, NamedStream):
+            self.filename = filename
+        else:
+            if hasattr(filename, 'topology'):
+                self.filename = filename
+            else:
+                self.filename = str(filename)
+                if "MMTF" in self.filename:
+                    self.filename = filename
 
         self.convert_units = convert_units