xmanager-slurm 0.4.10__py3-none-any.whl → 0.4.11__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.


This version of xmanager-slurm might be problematic. Click here for more details.

xm_slurm/config.py CHANGED
@@ -62,9 +62,14 @@ class SlurmSSHConfig:
62
62
  if self.host_public_key is None:
63
63
  return None
64
64
 
65
- return asyncssh.import_known_hosts(
66
- f"[{self.host}]:{self.port} {self.host_public_key.algorithm} {self.host_public_key.key}"
65
+ hostname = f"{self.host}"
66
+ if self.port is not None and self.port != asyncssh.DEFAULT_PORT:
67
+ hostname = f"[{hostname}]:{self.port}"
68
+
69
+ known_hosts = asyncssh.import_known_hosts(
70
+ f"{hostname} {self.host_public_key.algorithm} {self.host_public_key.key}"
67
71
  )
72
+ return known_hosts
68
73
 
69
74
  @functools.cached_property
70
75
  def config(self) -> asyncssh.config.SSHConfig:
@@ -117,9 +122,9 @@ class SlurmSSHConfig:
117
122
  def connection_options(self) -> asyncssh.SSHClientConnectionOptions:
118
123
  options = asyncssh.SSHClientConnectionOptions(
119
124
  config=None,
120
- kbdint_auth=False,
121
125
  disable_trivial_auth=True,
122
126
  known_hosts=self.known_hosts,
127
+ server_host_key_algs=self.host_public_key.algorithm if self.host_public_key else None,
123
128
  )
124
129
  options.prepare(last_config=self.config)
125
130
  return options
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: xmanager-slurm
3
- Version: 0.4.10
3
+ Version: 0.4.11
4
4
  Summary: Slurm backend for XManager.
5
5
  Project-URL: GitHub, https://github.com/jessefarebro/xm-slurm
6
6
  Author-email: Jesse Farebrother <jfarebro@cs.mcgill.ca>
@@ -1,6 +1,6 @@
1
1
  xm_slurm/__init__.py,sha256=WgRn9HDYa5H3sfIH-HZu33liBOh98jM4GqcR349RaSY,1086
2
2
  xm_slurm/batching.py,sha256=GbKBsNz9w8gIc2fHLZpslC0e4K9YUfLXFHmjduRRCfQ,4385
3
- xm_slurm/config.py,sha256=PvdLG6fSNfbABbtwELbnytx85vWLPtyVfEJtoQsLf94,7084
3
+ xm_slurm/config.py,sha256=i9WuxjfGBgVoHHDNk3BvO7LCwpBTJeRnOiFvTU-FHrk,7330
4
4
  xm_slurm/console.py,sha256=UpMqeJ0C8i0pkue1AHnnyyX0bFJ9zZeJ7HBR6yhuA8A,54
5
5
  xm_slurm/constants.py,sha256=zefVtlFdflgSolie5g_rVxWV-Zpydxapchm3y0a2FDc,999
6
6
  xm_slurm/dependencies.py,sha256=-5gN_tpfs3dOA7H5_MIHO2ratb7F5Pm_yjkR5rZcgI8,6421
@@ -44,8 +44,8 @@ xm_slurm/templates/slurm/fragments/monitor.bash.j2,sha256=BJ1brSjhESOe9VX_OYaPyy
44
44
  xm_slurm/templates/slurm/fragments/proxy.bash.j2,sha256=VJLglZo-Nvx9R-qe3rHTxr07CylTQ6Z9NwBzvIpAZrA,814
45
45
  xm_slurm/templates/slurm/runtimes/apptainer.bash.j2,sha256=lE2EWVCK2O-n08RL4_MJYIikVTvODjcYKuv7Eh73Q2w,1932
46
46
  xm_slurm/templates/slurm/runtimes/podman.bash.j2,sha256=3j7K5eyXt_WhXK0EoMlxnhlmFVJ2JyxRKbsMRaDqzSs,1148
47
- xmanager_slurm-0.4.10.dist-info/METADATA,sha256=jsnzz5ge4NpBIC_V3R3KYVm9AyG3cDH96YDUZ-2ZeDc,1043
48
- xmanager_slurm-0.4.10.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
49
- xmanager_slurm-0.4.10.dist-info/entry_points.txt,sha256=_HLGmLgxuQLOPmF2gOFYDVq2HqtMVD_SzigHvUh8TCY,49
50
- xmanager_slurm-0.4.10.dist-info/licenses/LICENSE.md,sha256=IxstXr3MPHwTJ5jMrByHrQsR1ZAGQ2U_uz_4qzI_15Y,11756
51
- xmanager_slurm-0.4.10.dist-info/RECORD,,
47
+ xmanager_slurm-0.4.11.dist-info/METADATA,sha256=f7oAbg0CGJLT3iQUERwjnB4IdhrDzgDoYEL2izYTA0o,1043
48
+ xmanager_slurm-0.4.11.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
49
+ xmanager_slurm-0.4.11.dist-info/entry_points.txt,sha256=_HLGmLgxuQLOPmF2gOFYDVq2HqtMVD_SzigHvUh8TCY,49
50
+ xmanager_slurm-0.4.11.dist-info/licenses/LICENSE.md,sha256=IxstXr3MPHwTJ5jMrByHrQsR1ZAGQ2U_uz_4qzI_15Y,11756
51
+ xmanager_slurm-0.4.11.dist-info/RECORD,,