MerchanBotCLI/utilities.py

# Create a list of hashmap key values .
import torch
import time
import gc
import os

# Utility Functions to be called from all modules

# Function to return a list of keys of a nested dictionary using it's key value (item or creature)
def keys_list(dict, index):
    keys_list=list(dict.keys())
    return keys_list[index]

def reclaim_mem():
    
    print(f"Memory before del {torch.cuda.memory_allocated()}")
    torch.cuda.ipc_collect()
    gc.collect()
    torch.cuda.empty_cache()
    time.sleep(0.01)
    print(f"Memory after del {torch.cuda.memory_allocated()}")

def del_object(object):
        del object
        gc.collect()

def directory_contents(directory_path):
    if os.path.isdir(directory_path) :
          contents = os.listdir(directory_path)
          return contents
    else : pass

def delete_files(file_paths):
     
    for file_path in file_paths:
        try:
            os.remove(file_path)
        except OSError as e:
            print(f"Error: {file_path} : {e.strerror}")
    file_paths.clear()