Commit 59c9c89e authored by Ambra Di Piano's avatar Ambra Di Piano
Browse files

add partition

parent 931f3b82

astrort/configure/check_configuration.py

+2 −1
Original line number Diff line number Diff line
@@ -52,7 +52,7 @@ class CheckConfiguration(): 
        return self

    

    def check_slurm(self):

        keys = ['nodes', 'tasks', 'cpus', 'mem', 'environment', 'name', 'account']

        keys = ['nodes', 'tasks', 'cpus', 'mem', 'environment', 'name', 'account', 'partition']

        assert self.conf['slurm'].keys() == keys

        assert type(self.conf['slurm']['nodes']) == int

        assert type(self.conf['slurm']['tasks']) == int
@@ -61,4 +61,5 @@ class CheckConfiguration(): 
        assert type(self.conf['slurm']['environment']) == str 

        assert type(self.conf['slurm']['name']) == str 

        assert type(self.conf['slurm']['account']) == str 

        assert type(self.conf['slurm']['partition']) == str 

        return self
 
 No newline at end of file