Use set() for Python 2.6 compat

Flamefire · Jul 22, 2020 · b7761e1 · b7761e1
1 parent 32d2fa1
commit b7761e1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/easybuild/easyblocks/p/pytorch.py b/easybuild/easyblocks/p/pytorch.py
@@ -84,7 +84,7 @@ def configure_step(self):
             options.append('BLAS=Eigen')
 
         known_dependencies = ('FFmpeg', 'gflags', 'glog')
-        dependency_names = {dep['name'] for dep in self.cfg.dependencies()}
+        dependency_names = set(dep['name'] for dep in self.cfg.dependencies())
         for dep in known_dependencies:
             if dep in dependency_names:
                 options.append('USE_%s=1' % dep.upper())