From daf022e3e7ed36d627faa54c436a9d6fa5327f21 Mon Sep 17 00:00:00 2001
From: Alexander Guschin <1aguschin@gmail.com>
Date: Wed, 1 Mar 2023 16:54:01 +0600
Subject: [PATCH] Convert requirement to bytes if needed (#622)

close #620
---
 mlem/contrib/docker/context.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/mlem/contrib/docker/context.py b/mlem/contrib/docker/context.py
index 0058ad30..185bf98c 100644
--- 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)