diff --git a/trains_agent/commands/worker.py b/trains_agent/commands/worker.py index 73ed5e8..72dcd3b 100644 --- a/trains_agent/commands/worker.py +++ b/trains_agent/commands/worker.py @@ -1693,7 +1693,7 @@ class Worker(ServiceCommandSection): def install_requirements_for_package_api( self, execution, repo_info, requirements_manager, cached_requirements=None, cwd=None, package_api=None, ): - # type: (ExecutionInfo, RepoInfo, RequirementsManager, Optional[dict]) -> None + # type: (ExecutionInfo, RepoInfo, RequirementsManager, Optional[dict], Optional[str], Optional[Any]) -> None """ :summary: Install requirements for task script using pip. :description: A file named "requirements.txt" is looked for in each containing folder between the @@ -1703,6 +1703,7 @@ class Worker(ServiceCommandSection): :param repo_info: repository information :param requirements_manager: requirements manager for task :param cached_requirements: cached requirements from previous run + :param cwd: current folder :param package_api: package_api to be used when installing requirements """ if package_api: diff --git a/trains_agent/helper/repo.py b/trains_agent/helper/repo.py index 4343704..4d24247 100644 --- a/trains_agent/helper/repo.py +++ b/trains_agent/helper/repo.py @@ -254,6 +254,8 @@ class VCS(object): parsed_url.username = "git" parsed_url.password = None # make sure there is no port in the final url (safe_furl support) + # the original port was an https port, and we do not know if there is a different ssh port, + # so we have to clear the original port specified (https) and use the default ssh schema port. parsed_url.port = None url = parsed_url.url return url