Convert requirement to bytes if needed (#622)

close #620
iterative · Mar 1, 2023 · daf022e · daf022e
1 parent c448416
commit daf022e
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/mlem/contrib/docker/context.py b/mlem/contrib/docker/context.py
@@ -398,6 +398,8 @@ def write_local_sources(self, requirements: Requirements):
             logger.debug('Putting model source "%s" to distribution...', path)
             full_path = posixpath.join(self.path, path)
             self.fs.makedirs(posixpath.dirname(full_path), exist_ok=True)
+            if isinstance(src, str):  # type: ignore
+                src = src.encode()  # type: ignore
             with self.fs.open(full_path, "wb") as f:
                 f.write(src)