Release: v0.2.1

docs/source/conf.py

@@ -25,7 +25,7 @@ copyright = "2020, The Hugging Face Team, Licenced under the Apache License, Ver
 author = "huggingface"
 
 # The short X.Y version
-version = "0.2.0"
+version = "0.2.1"
 
 # -- General configuration ---------------------------------------------------
 

setup.py

@@ -37,7 +37,7 @@ extras["sagemaker"] = [
 
 setup(
     name="accelerate",
-    version="0.2.0",
+    version="0.2.1",
     description="Accelerate",
     long_description=open("README.md", "r", encoding="utf-8").read(),
     long_description_content_type="text/markdown",

src/accelerate/__init__.py

@@ -2,7 +2,7 @@
 # There's no way to ignore "F401 '...' imported but unused" warnings in this
 # module, but to preserve other warnings. So, don't check this module at all.
 
-__version__ = "0.2.0"
+__version__ = "0.2.1"
 
 from .accelerator import Accelerator
 from .kwargs_handlers import DistributedDataParallelKwargs, GradScalerKwargs

src/accelerate/commands/config/config_args.py

@@ -44,7 +44,7 @@ def load_config_from_file(config_file):
         if config_file.endswith(".json"):
             if (
                 json.load(f).get("compute_environment", ComputeEnvironment.LOCAL_MACHINE)
-                is ComputeEnvironment.LOCAL_MACHINE
+                == ComputeEnvironment.LOCAL_MACHINE
             ):
                 config_class = ClusterConfig
             else:
@@ -53,7 +53,7 @@ def load_config_from_file(config_file):
         else:
             if (
                 yaml.safe_load(f).get("compute_environment", ComputeEnvironment.LOCAL_MACHINE)
-                is ComputeEnvironment.LOCAL_MACHINE
+                == ComputeEnvironment.LOCAL_MACHINE
             ):
                 config_class = ClusterConfig
             else: