code: def get_compatible_batch_size(batch_size, neg_sample_size):
      if (
          neg_sample_size < batch_size
          and batch_size % neg_sample_size != 0
      ):
          old_batch_size = batch_size
          batch_size = int(
              math.ceil(batch_size / neg_sample_size) * neg_sample_size
          )
          print(
              "batch size ({}) is incompatible to the negative sample size ({}). Change the batch size to {}".format(
                  old_batch_size, neg_sample_size, batch_size
              )
          )
      return batch_size
  
file_path: ogb/examples/lsc/wikikg90m/dgl-ke-ogb-lsc/python/dglke/utils.py
imports: import math
  import os
  import csv
  import argparse
  import json
  import numpy
label_desc: Use image node_id 1 for calling a global function with example usage: get_compatible_batch_size(batch_size, neg_sample_size) and returns: batch_size