add ability to connect a neptune.ai run (huggingface#13319)

fcakyon · web-flow · commit 6f3c99acca30 · 2021-08-30T09:59:17.000-04:00
when `NEPTUNE_RUN_ID` environmetnt variable is set, neptune will log into the previous run with id `NEPTUNE_RUN_ID`
diff --git a/src/transformers/integrations.py b/src/transformers/integrations.py
@@ -728,6 +728,7 @@ def setup(self, args, state, model):
                 api_token=os.getenv("NEPTUNE_API_TOKEN"),
                 mode=os.getenv("NEPTUNE_CONNECTION_MODE", "async"),
                 name=os.getenv("NEPTUNE_RUN_NAME", None),
+                run=os.getenv("NEPTUNE_RUN_ID", None),
             )
             combined_dict = args.to_dict()
             if hasattr(model, "config") and model.config is not None:

Original file line number	Diff line number	Diff line change
`@@ -728,6 +728,7 @@ def setup(self, args, state, model):`
`728`	`728`	`api_token=os.getenv("NEPTUNE_API_TOKEN"),`
`729`	`729`	`mode=os.getenv("NEPTUNE_CONNECTION_MODE", "async"),`
`730`	`730`	`name=os.getenv("NEPTUNE_RUN_NAME", None),`
	`731`	`+ run=os.getenv("NEPTUNE_RUN_ID", None),`
`731`	`732`	`)`
`732`	`733`	`combined_dict = args.to_dict()`
`733`	`734`	`if hasattr(model, "config") and model.config is not None:`