JBAujogue commited on
Commit
f14f9ff
·
1 Parent(s): a478bce

extend line length in ruff formatting

Browse files
Files changed (3) hide show
  1. pyproject.toml +3 -0
  2. src/rubik/__main__.py +1 -1
  3. src/rubik/cube.py +4 -12
pyproject.toml CHANGED
@@ -26,3 +26,6 @@ torch = { index = "torch-cu126" }
26
  [[tool.uv.index]]
27
  name = "torch-cu126"
28
  url = "https://download.pytorch.org/whl/cu126"
 
 
 
 
26
  [[tool.uv.index]]
27
  name = "torch-cu126"
28
  url = "https://download.pytorch.org/whl/cu126"
29
+
30
+ [tool.ruff]
31
+ line-length = 120
src/rubik/__main__.py CHANGED
@@ -3,4 +3,4 @@ from fire import Fire
3
  from rubik.hello import hello_world
4
 
5
 
6
- Fire({'hello': hello_world})
 
3
  from rubik.hello import hello_world
4
 
5
 
6
+ Fire({"hello": hello_world})
src/rubik/cube.py CHANGED
@@ -26,9 +26,7 @@ class Cube:
26
  cube = Cube.from_default(['U', 'L', 'C', 'R', 'B', 'D'], size = 3)
27
  """
28
  assert (num := len(set(colors))) == 6, f"Expected 6 distinct colors, got {num}"
29
- assert isinstance(size, int) and size > 1, (
30
- f"Expected non-zero integrer size, got {size}"
31
- )
32
 
33
  # build tensor filled with 0's, and fill the faces with 1's
34
  n = size - 1
@@ -54,10 +52,7 @@ class Cube:
54
  self.tensor[:, n, :, 4, :].argmax(dim=-1), # back
55
  self.tensor[:, :, 0, 5, :].argmax(dim=-1), # down
56
  ]
57
- return [
58
- [[self.colors[i - 1] for i in row] for row in face.tolist()]
59
- for face in grid
60
- ]
61
 
62
  def __str__(self):
63
  """
@@ -79,10 +74,7 @@ class Cube:
79
  void = " " * self.size
80
  top = "\n".join(" ".join([void, "".join(row), void, void]) for row in grid[0])
81
  middle = "\n".join(
82
- " ".join("".join(grid[face_i][row_i]) for face_i in range(1, 5))
83
- for row_i in range(self.size)
84
- )
85
- bottom = "\n".join(
86
- " ".join((void, "".join(row), void, void)) for row in grid[-1]
87
  )
 
88
  return "\n".join([top, middle, bottom])
 
26
  cube = Cube.from_default(['U', 'L', 'C', 'R', 'B', 'D'], size = 3)
27
  """
28
  assert (num := len(set(colors))) == 6, f"Expected 6 distinct colors, got {num}"
29
+ assert isinstance(size, int) and size > 1, f"Expected non-zero integrer size, got {size}"
 
 
30
 
31
  # build tensor filled with 0's, and fill the faces with 1's
32
  n = size - 1
 
52
  self.tensor[:, n, :, 4, :].argmax(dim=-1), # back
53
  self.tensor[:, :, 0, 5, :].argmax(dim=-1), # down
54
  ]
55
+ return [[[self.colors[i - 1] for i in row] for row in face.tolist()] for face in grid]
 
 
 
56
 
57
  def __str__(self):
58
  """
 
74
  void = " " * self.size
75
  top = "\n".join(" ".join([void, "".join(row), void, void]) for row in grid[0])
76
  middle = "\n".join(
77
+ " ".join("".join(grid[face_i][row_i]) for face_i in range(1, 5)) for row_i in range(self.size)
 
 
 
 
78
  )
79
+ bottom = "\n".join(" ".join((void, "".join(row), void, void)) for row in grid[-1])
80
  return "\n".join([top, middle, bottom])