GeneInterpolate not accepting border_predictions

[HDictus]

I am getting this traceback when running full_pipeline.py

Traceback (most recent call last):
  File "/gpfs/bbp.cscs.ch/project/proj148/atlas/Deep-Atlas/pipeline/full_pipeline.py", line 253, in <module>
    sys.exit(main(**kwargs))
  File "/gpfs/bbp.cscs.ch/project/proj148/atlas/Deep-Atlas/pipeline/full_pipeline.py", line 240, in main
    output_dir=interpolation_results_dir,
  File "/gpfs/bbp.cscs.ch/project/proj148/atlas/Deep-Atlas/pipeline/interpolate_gene.py", line 186, in main
    gene_dataset, interpolator_model, border_predictions=False
TypeError: __init__() got an unexpected keyword argument 'border_predictions'

I have the latest version of atlas-interpolator installed, but I guess it's using a version it installed inside the singularity environment? (I have no Idea how this works). If so, then the issue must be that that is out of date.