Update utils.py to support Windows (#298)

alex-petrenko · Aug 28, 2024 · 066b571 · 066b571
1 parent bb0a6ac
commit 066b571
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/sample_factory/utils/utils.py b/sample_factory/utils/utils.py
@@ -5,7 +5,7 @@
 import logging
 import operator
 import os
-import pwd
+import getpass
 import tempfile
 import time
 from os.path import join
@@ -385,9 +385,8 @@ def remove_if_exists(file):
 
 
 def get_username():
-    uid = os.getuid()
     try:
-        return pwd.getpwuid(uid).pw_name
+        return getpass.getuser()
     except KeyError:
         # worst case scenario - let's just use uid
         return str(uid)