import sys
import time
from random import shuffle
from FairDistributor import FairDistributor


def main():
    # User input for the number of targets and objects.
    number_of_targets = int(sys.argv[1])
    number_of_objects = int(sys.argv[2])
    # Generate dummy lists for objects, targets and dummy matrix for weights
    targets = ['target_{0}'.format(str(s))
               for s in range(number_of_targets)]
    objects = ['object_{0}'.format(str(s))
               for s in range(number_of_objects)]
    dummy_weights = list(range(1, number_of_objects+1))
    weights_matrix = list()
    for _ in range(number_of_targets):
        new_random_weight_list = list(dummy_weights)
        shuffle(new_random_weight_list)
        weights_matrix.append(new_random_weight_list)
    start_time = time.time()
    distributor = FairDistributor(targets, objects, weights_matrix)
    distributor.distribute()
    elapsed_time = time.time() - start_time
    print('Number of Targets: {0}\nNumber of Objects: {1}\nTime elapsed: {2}'.format(
        number_of_targets, number_of_objects, elapsed_time))

if __name__ == '__main__':
    main()

import sys
import time
from random import shuffle
from vania.fair_distributor import FairDistributor


def main():
    # User input for the number of targets and objects.
    number_of_targets = int(sys.argv[1])
    number_of_objects = int(sys.argv[2])

    # Generate dummy lists for objects, targets and dummy matrix for weights
    targets = ['target_{0}'.format(str(s))
               for s in range(number_of_targets)]
    objects = ['object_{0}'.format(str(s))
               for s in range(number_of_objects)]
    dummy_weights = list(range(1, number_of_objects+1))
    weights_matrix = list()
    for _ in range(number_of_targets):
        new_random_weight_list = list(dummy_weights)
        shuffle(new_random_weight_list)
        weights_matrix.append(new_random_weight_list)

    # Benchmark solver
    start_time = time.time()
    distributor = FairDistributor(targets, objects, weights_matrix)
    distributor.distribute()
    elapsed_time = time.time() - start_time

    # Output
    print('Number of Targets: {0}\nNumber of Objects: {1}\nTime elapsed: {2}'.format(
        number_of_targets, number_of_objects, elapsed_time))

if __name__ == '__main__':
    main()

Reformat random generator reformat code
Reformat random generator reformat code

  import sys
  import time
  from random import shuffle
- from FairDistributor import FairDistributor
+ from vania.fair_distributor import FairDistributor
  
  
  def main():
      # User input for the number of targets and objects.
      number_of_targets = int(sys.argv[1])
      number_of_objects = int(sys.argv[2])
+ 
      # Generate dummy lists for objects, targets and dummy matrix for weights
      targets = ['target_{0}'.format(str(s))
                 for s in range(number_of_targets)]
      objects = ['object_{0}'.format(str(s))
                 for s in range(number_of_objects)]
      dummy_weights = list(range(1, number_of_objects+1))
      weights_matrix = list()
      for _ in range(number_of_targets):
          new_random_weight_list = list(dummy_weights)
          shuffle(new_random_weight_list)
          weights_matrix.append(new_random_weight_list)
+ 
+     # Benchmark solver
      start_time = time.time()
      distributor = FairDistributor(targets, objects, weights_matrix)
      distributor.distribute()
      elapsed_time = time.time() - start_time
+ 
+     # Output
      print('Number of Targets: {0}\nNumber of Objects: {1}\nTime elapsed: {2}'.format(
          number_of_targets, number_of_objects, elapsed_time))
  
  if __name__ == '__main__':
      main()
  
Reformat random generator reformat code
## Code Before:
import sys
import time
from random import shuffle
from FairDistributor import FairDistributor


def main():
    # User input for the number of targets and objects.
    number_of_targets = int(sys.argv[1])
    number_of_objects = int(sys.argv[2])
    # Generate dummy lists for objects, targets and dummy matrix for weights
    targets = ['target_{0}'.format(str(s))
               for s in range(number_of_targets)]
    objects = ['object_{0}'.format(str(s))
               for s in range(number_of_objects)]
    dummy_weights = list(range(1, number_of_objects+1))
    weights_matrix = list()
    for _ in range(number_of_targets):
        new_random_weight_list = list(dummy_weights)
        shuffle(new_random_weight_list)
        weights_matrix.append(new_random_weight_list)
    start_time = time.time()
    distributor = FairDistributor(targets, objects, weights_matrix)
    distributor.distribute()
    elapsed_time = time.time() - start_time
    print('Number of Targets: {0}\nNumber of Objects: {1}\nTime elapsed: {2}'.format(
        number_of_targets, number_of_objects, elapsed_time))

if __name__ == '__main__':
    main()

## Instruction:
Reformat random generator reformat code
## Code After:
import sys
import time
from random import shuffle
from vania.fair_distributor import FairDistributor


def main():
    # User input for the number of targets and objects.
    number_of_targets = int(sys.argv[1])
    number_of_objects = int(sys.argv[2])

    # Generate dummy lists for objects, targets and dummy matrix for weights
    targets = ['target_{0}'.format(str(s))
               for s in range(number_of_targets)]
    objects = ['object_{0}'.format(str(s))
               for s in range(number_of_objects)]
    dummy_weights = list(range(1, number_of_objects+1))
    weights_matrix = list()
    for _ in range(number_of_targets):
        new_random_weight_list = list(dummy_weights)
        shuffle(new_random_weight_list)
        weights_matrix.append(new_random_weight_list)

    # Benchmark solver
    start_time = time.time()
    distributor = FairDistributor(targets, objects, weights_matrix)
    distributor.distribute()
    elapsed_time = time.time() - start_time

    # Output
    print('Number of Targets: {0}\nNumber of Objects: {1}\nTime elapsed: {2}'.format(
        number_of_targets, number_of_objects, elapsed_time))

if __name__ == '__main__':
    main()
