From a71370a756fcf31babe8f72e30991608a8b05780 Mon Sep 17 00:00:00 2001
From: Han Xiao <han.xiao@jina.ai>
Date: Sat, 6 Aug 2022 00:41:00 +0200
Subject: [PATCH] fix: float to int in seeding (#141)

---
 discoart/config.py | 10 +++-------
 1 file changed, 3 insertions(+), 7 deletions(-)

diff --git a/discoart/config.py b/discoart/config.py
index 5738fd5..3083087 100644
--- a/discoart/config.py
+++ b/discoart/config.py
@@ -53,21 +53,17 @@ def load_config(
         cfg.update(**user_config)
 
     int_keys = {k for k, v in default_args.items() if isinstance(v, int)}
-    int_keys.union(
-        {
-            'seed',
-        }
-    )
+    int_keys.add('seed')
 
     for k, v in cfg.items():
-        if k in int_keys and isinstance(v, float):
+        if k in int_keys and v is not None and not isinstance(v, int):
             cfg[k] = int(v)
         if k == 'width_height':
             cfg[k] = [int(vv) for vv in v]
 
     cfg.update(
         **{
-            'seed': cfg['seed'] or random.randint(0, 2**32),
+            'seed': int(cfg['seed'] or random.randint(0, 2**32)),
         }
     )